-
-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathsolution.cpp
84 lines (68 loc) · 1.97 KB
/
solution.cpp
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode() : val(0), left(nullptr), right(nullptr) {}
* TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
* TreeNode(int x, TreeNode *left, TreeNode *right) : val(x), left(left), right(right) {}
* };
*/
/*
Complexity Analysis: DFS Solution
Time - O(N)
Space - O(N)
*/
class Solution {
void dfs(TreeNode* root, int level, vector<vector<int>>& values) {
if(root == NULL) return;
if(level == (int)values.size()) values.push_back({});
values[level].push_back(root->val);
dfs(root->left, level + 1, values);
dfs(root->right, level + 1, values);
}
public:
vector<vector<int>> levelOrderBottom(TreeNode* root) {
if(root == NULL) return {};
vector<vector<int>> values;
dfs(root, 0, values);
reverse(values.begin(), values.end());
return values;
}
};
/*
Complexity Analysis: BFS Solution
Time - O(N)
Space - O(N)
*/
class Solution {
public:
vector<vector<int>> levelOrderBottom(TreeNode* root) {
if(root == NULL) return {};
vector<vector<int>> values;
vector<int> level;
queue<TreeNode*> q;
q.emplace(root);
int n = q.size();
while(!q.empty()) {
TreeNode* current = q.front();
q.pop();
level.push_back(current->val);
--n;
if(current->left) {
q.emplace(current->left);
}
if(current->right) {
q.emplace(current->right);
}
if(n == 0) {
n = q.size();
values.push_back(level);
level = {};
}
}
reverse(values.begin(), values.end());
return values;
}
};