LeetCode103. 二叉树的锯齿形层次遍历

Ember_Sky

于 2020-10-29 22:53:00 发布

阅读量212

点赞数

分类专栏： # LeetCode 文章标签：二叉树队列数据结构算法

本文链接：https://blog.youkuaiyun.com/wzh1378008099/article/details/109373665

版权

LeetCode 专栏收录该内容

111 篇文章

订阅专栏

这篇博客介绍了如何使用双端队列（deque）实现二叉树的锯齿形层次遍历。作者通过维护两个队列，分别处理奇数层和偶数层的节点，从不同方向取出节点，达到锯齿形的效果。代码简洁明了，适合初学者理解二叉树遍历和双端队列的应用。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

//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;
  }
};