当前位置:网站首页>JZ59.按之字型顺序打印二叉树

JZ59.按之字型顺序打印二叉树

2022-06-21 16:53:00 SS_zico

牛客:JZ59
题目描述
请实现一个函数按照之字形打印二叉树,即第一行按照从左到右的顺序打印,第二层按照从右至左的顺序打印,第三行按照从左到右的顺序打印,其他行以此类推。

示例1
输入
{8,6,10,5,7,9,11}
返回值
[[8],[10,6],[5,7,9,11]]

class Solution {
    
public:
    vector<vector<int> > Print(TreeNode* pRoot) {
    
        vector<vector<int>> ret;
        if (!pRoot) return ret;
        queue<TreeNode*> q;
        q.push(pRoot);
        int level = 0;

        while (!q.empty()) {
    
            int sz = q.size();
            vector<int> ans;
            while (sz--) {
    
                TreeNode *node = q.front();
                q.pop();
                ans.push_back(node->val);

                if (node->left) q.push(node->left);
                if (node->right) q.push(node->right);
            }
            ++level;
            if (!(level&1)) // 偶数层 反转一下
                reverse(ans.begin(), ans.end());
            ret.push_back(ans);
        }
        return ret;
    }

};
原网站

版权声明
本文为[SS_zico]所创,转载请带上原文链接,感谢
https://blog.csdn.net/GHR7719/article/details/113644169