//103. 二叉树的锯齿形层次遍历
/*
* 维护两个数组
* 一个专门记录奇数层的节点,另一个专门记录双数层的节点
* 一个从头部开始取出节点,另一个从尾部开始取出节点
* 一个将节点放进头部,另一个将节点放在尾部
* 代码用的是双向队列,但是写题解的时候才发现其实算是栈
* 还算简单
*/
class Solution {
public:
vector<vector<int>> zigzagLevelOrder(TreeNode* root) {
vector<vector<int>>ans;
if (root == NULL) return ans;
deque<TreeNode*> dq1;
deque<TreeNode*> dq2;
dq1.emplace_back(root);
vector<int>t;
while (dq1.size() || dq2.size()) {
while (dq1.size()) {
auto k = dq1.front(); dq1.pop_front();
t.emplace_back(k->val);
if (k->left != NULL) dq2.emplace_back(k->left);
if (k->right != NULL) dq2.emplace_back(k->right);
}
if (t.size()) ans.emplace_back(t);
t.clear();
while (dq2.size()) {
auto k = dq2.back(); dq2.pop_back();
t.emplace_back(k->val);
if (k->right != NULL) dq1.push_front(k->right);
if (k->left != NULL) dq1.push_front(k->left);
}
if (t.size()) ans.emplace_back(t);
t.clear();
}
return ans;
}
};
LeetCode103. 二叉树的锯齿形层次遍历
最新推荐文章于 2022-12-23 15:56:15 发布