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

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

©️2020 CSDN 皮肤主题: 技术黑板 设计师:CSDN官方博客 返回首页