|
1 | 1 | /*
|
2 | 2 | Author: Annie Kim, anniekim.pku@gmail.com
|
3 | 3 | Date: May 16, 2013
|
4 |
| - Update: Aug 17, 2013 |
| 4 | + Update: Sep 12, 2013 |
5 | 5 | Problem: Binary Tree Zigzag Level Order Traversal
|
6 | 6 | Difficulty: Easy
|
7 | 7 | Source: http://leetcode.com/onlinejudge#question_103
|
|
22 | 22 | [15,7]
|
23 | 23 | ]
|
24 | 24 |
|
25 |
| - Solution: 1. BFS(queue) + reverse vector. |
26 |
| - 2. Two stack. |
| 25 | + Solution: 1. Queue + reverse. |
| 26 | + 2. Two stacks. |
| 27 | + 3. Vector. Contributed by yinlinglin. |
27 | 28 | */
|
28 | 29 |
|
29 | 30 | /**
|
|
39 | 40 | class Solution {
|
40 | 41 | public:
|
41 | 42 | vector<vector<int>> zigzagLevelOrder(TreeNode *root) {
|
42 |
| - return zigzagLevelOrder_2(root); |
| 43 | + return zigzagLevelOrder_1(root); |
43 | 44 | }
|
44 | 45 |
|
| 46 | + // solution 1: Queue + Reverse. |
45 | 47 | vector<vector<int>> zigzagLevelOrder_1(TreeNode *root) {
|
46 | 48 | vector<vector<int>> res;
|
47 | 49 | if (!root) return res;
|
48 | 50 | queue<TreeNode *> q;
|
49 | 51 | q.push(root);
|
50 | 52 | q.push(NULL); // end indicator of one level
|
51 | 53 | bool left2right = true;
|
52 |
| - int level = 0; |
53 |
| - while (!q.empty()) |
| 54 | + vector<int> level; |
| 55 | + while (true) |
54 | 56 | {
|
55 |
| - TreeNode *front = q.front(); |
56 |
| - q.pop(); |
57 |
| - if (front) |
| 57 | + TreeNode *node = q.front(); q.pop(); |
| 58 | + if (node) |
58 | 59 | {
|
59 |
| - if (res.size() == level) |
60 |
| - res.push_back(vector<int>()); |
61 |
| - res[level].push_back(front->val); |
62 |
| - if (front->left) |
63 |
| - q.push(front->left); |
64 |
| - if (front->right) |
65 |
| - q.push(front->right); |
| 60 | + level.push_back(node->val); |
| 61 | + if (node->left) q.push(node->left); |
| 62 | + if (node->right) q.push(node->right); |
66 | 63 | }
|
67 | 64 | else
|
68 | 65 | {
|
69 |
| - if (!q.empty()) // CAUTIOUS! infinite loop |
70 |
| - q.push(NULL); |
71 |
| - if (!left2right) |
72 |
| - reverse(res[level].begin(), res[level].end()); |
| 66 | + if (!left2right) |
| 67 | + reverse(level.begin(), level.end()); |
| 68 | + res.push_back(level); |
| 69 | + level.clear(); |
| 70 | + if (q.empty()) break; |
| 71 | + q.push(NULL); |
73 | 72 | left2right = !left2right;
|
74 |
| - level++; |
75 | 73 | }
|
76 | 74 | }
|
77 | 75 | return res;
|
78 | 76 | }
|
79 | 77 |
|
| 78 | + // Solution 2: Two stacks. |
80 | 79 | vector<vector<int>> zigzagLevelOrder_2(TreeNode *root) {
|
81 | 80 | vector<vector<int>> res;
|
82 | 81 | if (!root) return res;
|
83 | 82 | stack<TreeNode *> stk[2];
|
84 |
| - bool left2right = false; |
| 83 | + bool left2right = true; |
85 | 84 | int cur = 1, last = 0;
|
86 | 85 | stk[last].push(root);
|
87 |
| - int level = 0; |
| 86 | + vector<int> level; |
88 | 87 | while (!stk[last].empty())
|
89 | 88 | {
|
90 |
| - TreeNode *node = stk[last].top(); |
| 89 | + TreeNode *node = stk[last].top(); |
91 | 90 | stk[last].pop();
|
92 | 91 | if (node)
|
93 | 92 | {
|
94 |
| - if (res.size() == level) |
95 |
| - res.push_back(vector<int>()); |
96 |
| - res[level].push_back(node->val); |
| 93 | + level.push_back(node->val); |
97 | 94 | if (left2right)
|
98 | 95 | {
|
99 |
| - stk[cur].push(node->right); |
100 | 96 | stk[cur].push(node->left);
|
| 97 | + stk[cur].push(node->right); |
101 | 98 | }
|
102 | 99 | else
|
103 | 100 | {
|
104 |
| - stk[cur].push(node->left); |
105 | 101 | stk[cur].push(node->right);
|
| 102 | + stk[cur].push(node->left); |
106 | 103 | }
|
107 | 104 | }
|
108 | 105 | if (stk[last].empty())
|
109 | 106 | {
|
110 |
| - cur = !cur; |
111 |
| - last = !last; |
| 107 | + if (!level.empty()) |
| 108 | + res.push_back(level); |
| 109 | + level.clear(); |
| 110 | + swap(cur, last); |
112 | 111 | left2right = !left2right;
|
113 |
| - level++; |
114 | 112 | }
|
115 | 113 | }
|
116 | 114 | return res;
|
117 | 115 | }
|
| 116 | + |
| 117 | + // Solution 3: Vector. Contributed by yinlinglin. |
| 118 | + // Compared to solution 1&2, this solution costs a little more space. |
| 119 | + // This solution uses only one single vector instead of two stacks in solution 2. |
| 120 | + vector<vector<int>> zigzagLevelOrder_3(TreeNode *root) { |
| 121 | + vector<vector<int>> result; |
| 122 | + if(!root) return result; |
| 123 | + vector<TreeNode*> v; |
| 124 | + v.push_back(root); |
| 125 | + bool left2right = true; |
| 126 | + int begin = 0, end = 0; |
| 127 | + while(begin <= end) |
| 128 | + { |
| 129 | + vector<int> row; |
| 130 | + for (int i = end; i >= begin; --i) |
| 131 | + { |
| 132 | + if (!v[i]) continue; |
| 133 | + row.push_back(v[i]->val); |
| 134 | + if(left2right) |
| 135 | + { |
| 136 | + v.push_back(v[i]->left); |
| 137 | + v.push_back(v[i]->right); |
| 138 | + } |
| 139 | + else |
| 140 | + { |
| 141 | + v.push_back(v[i]->right); |
| 142 | + v.push_back(v[i]->left); |
| 143 | + } |
| 144 | + } |
| 145 | + if (!row.empty()) |
| 146 | + result.push_back(row); |
| 147 | + begin = end + 1; |
| 148 | + end = v.size() - 1; |
| 149 | + left2right = !left2right; |
| 150 | + } |
| 151 | + return result; |
| 152 | + } |
118 | 153 | };
|
0 commit comments