剑指 offer acwing 45 之字形打印二叉树

最新推荐文章于 2022-08-31 18:49:29 发布

(xsj)

最新推荐文章于 2022-08-31 18:49:29 发布

阅读量104

点赞数

分类专栏： # 二叉树与二分搜索树

本文链接：https://blog.csdn.net/qq_44791484/article/details/115150144

版权

二叉树分层遍历广度优先搜索节点翻转 C++

关键词由CSDN通过智能技术生成

二叉树与二分搜索树专栏收录该内容

14 篇文章 0 订阅

订阅专栏

题面

在这里插入图片描述

题解

和44题一样，在每层末尾加一个nullptr ，就可以实现分层，然后加一个计数器，当cnt为奇数时，直接放入res，当cnt为偶数时，将cur翻转，然后放入res

代码

/**
 * Definition for a binary tree node.
 * struct TreeNode {
 *     int val;
 *     TreeNode *left;
 *     TreeNode *right;
 *     TreeNode(int x) : val(x), left(NULL), right(NULL) {}
 * };
 */
class Solution {
public:
    vector<vector<int>> printFromTopToBottom(TreeNode *root) {
        vector<vector<int>> res;
        if (!root) return res;
        vector<int> cur;
        queue<TreeNode *> q;
        q.push(root);
        q.push(nullptr);
        int cnt = 1;
        while (q.size()) {
            auto t = q.front();
            q.pop();
            if (t) {
                cur.push_back(t->val);
                if (t->left) q.push(t->left);
                if (t->right) q.push(t->right);
            } else {
                if (cnt & 1) res.push_back(cur);
                else {
                    reverse(cur.begin(), cur.end());
                    res.push_back(cur);
                }
                cnt++;
                cur.clear();
                if (q.size()) q.push(nullptr);
            }
        }
        return res;
    }
};