剑指 32_3 之字形打印二叉树

剑指 32_3 之字形打印二叉树

全部刷题与学习记录

【C++刷题学习笔记目录】

【C++百万并发网络通信-笔记目录】

原题目

题目地址：剑指 Offer 32 - III. 从上到下打印二叉树 III

请实现一个函数按照之字形顺序打印二叉树，即第一行按照从左到右的顺序打印，第二层按照从右到左的顺序打印，第三行再按照从左到右的顺序打印，其他行以此类推。

例如:
给定二叉树: [3,9,20,null,null,15,7],

    3
   / \
  9  20
    /  \
   15   7

返回其层次遍历结果：

[
  [3],
  [20,9],
  [15,7]
]

---

考查知识点

二叉树、栈

---

自己的第一遍解法

既然是反转顺序，自然想到了栈的后入先出特性，只是没想到需要用两个栈。

---

好的解法

剑指上就给出了两个栈的配合反转，比如示例中将3存入栈0的时候，要将下一层节点9和20存入栈中，根据栈后入先出的规矩，我们要先输出20，那么就要最后保存20，那么也就是在第2层上我们要先保存左子节点，再保存右子节点。假设现在到了保存第3层子节点了，因为题目要最后输出7，那么我们就要首先保存7，就是先保存右子节点，再保存左子节点。因此，不同层的输出顺序就变成了入栈的左右子节点顺序问题。

struct TreeNode {
    int val;
    TreeNode *left;
    TreeNode *right;
    TreeNode(int x) : val(x), left(nullptr), right(nullptr) {}
};

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        if (!root)
            return vector<vector<int>> {};

        vector<vector<int>> res;
        vector<int> tmpRes;
        stack<TreeNode*> level[2];
        int cur = 0;//当前处理栈在数组中的下标,==0：当前偶数层；==1：当前为奇数层
        int next = 1;

        level[cur].push(root);
        while (!level[cur].empty() || !level[next].empty())
        {
            TreeNode* node = level[cur].top();
            level[cur].pop();
            tmpRes.push_back(node->val);

            if (cur == 0)
            {
                if (node->left)
                    level[next].push(node->left);
                if (node->right)
                    level[next].push(node->right);
            }
            else
            {
                if (node->right)
                    level[next].push(node->right);
                if (node->left)
                    level[next].push(node->left);
            }

            if (level[cur].empty())
            {
                res.push_back(tmpRes);
                tmpRes = {};
                cur = 1 - cur;
                next = 1 - next;
            }
        }
        return res;
    }
};