forked from pezy/LeetCode
-
Notifications
You must be signed in to change notification settings - Fork 0
/
solution.h
36 lines (33 loc) · 916 Bytes
/
solution.h
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
#include <vector>
#include <queue>
#include <cstddef>
using std::vector;
struct TreeNode {
int val;
TreeNode *left;
TreeNode *right;
TreeNode(int x) : val(x), left(NULL), right(NULL) {}
};
class Solution {
public:
vector<vector<int> > zigzagLevelOrder(TreeNode *root) {
vector<vector<int>> retv;
if (!root) return retv;
std::queue<TreeNode*> q;
q.push(root);
bool reverse = false;
vector<int> v;
for (int cur_level_sz = 1, next_level_sz = 0; !q.empty(); ) {
TreeNode *node = q.front(); q.pop();
v.push_back(node->val); --cur_level_sz;
if (node->left) { q.push(node->left); ++next_level_sz; }
if (node->right) { q.push(node->right); ++next_level_sz; }
if (!cur_level_sz) {
retv.push_back(reverse ? vector<int>(v.crbegin(), v.crend()) : v); v.clear();
cur_level_sz = next_level_sz; next_level_sz = 0;
reverse = !reverse;
}
}
return retv;
}
};