剑指 Offer 32 - III. 从上到下打印二叉树 III

最新推荐文章于 2024-07-14 14:13:59 发布

原创最新推荐文章于 2024-07-14 14:13:59 发布 · 185 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#二叉树

剑指offer 专栏收录该内容

34 篇文章

订阅专栏

本文介绍了一种使用广度优先搜索(BFS)策略按之字形顺序打印二叉树的方法。通过维护一个标志变量来切换每一行的左右顺序，实现了从上到下交替左右打印的效果。给定一棵包含3个节点的二叉树，代码展示了如何返回按之字形顺序排列的节点值列表。

剑指 Offer 32 - III. 从上到下打印二叉树 III

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

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

    3
   / \
  9  20
    /  \
   15   7
返回：
[
  [3],
  [20,9],
  [15,7]
]

限制：

节点总数 <= 1000

思路：BFS + 标记

class Solution {
public:
    vector<vector<int>> levelOrder(TreeNode* root) {
        vector<vector<int>> ans;
        queue<TreeNode*> q;
        if(!root) return {};
        q.push(root);
        int f = 1;
        while(q.size()){
            auto n = q.size();
            vector<int> v;
            while(n--){
                auto t = q.front(); q.pop();
                v.push_back(t->val);
                if(t->left) q.push(t->left);
                if(t->right) q.push(t->right);
            }
            if(!f){
                reverse(v.begin(),v.end());
            } 
            f ^= 1;
            ans.push_back(v);
        }
        return ans;
    }
};