剑指offer---把二叉树打印成多行

最新推荐文章于 2022-03-09 09:27:47 发布

谛听-

最新推荐文章于 2022-03-09 09:27:47 发布

阅读量154

点赞数

CC 4.0 BY-SA版权

分类专栏：题树

本文链接：https://blog.youkuaiyun.com/u012319493/article/details/79515405

题同时被 2 个专栏收录

456 篇文章

订阅专栏

树

46 篇文章

订阅专栏

本文介绍了一种算法，用于从上至下按层打印二叉树，并在同一层内从左至右输出结点值。该算法使用两个队列来实现逐层遍历，适用于数据结构与算法学习。

题目描述
从上到下按层打印二叉树，同一层结点从左至右输出。每一层输出一行。

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
class Solution {
public:
        vector<vector<int> > Print(TreeNode* pRoot) 
        {
            vector<vector<int> > ans;
            vector<int> tmp;
            if(!pRoot)
                return ans;

            queue<TreeNode*> q;
            queue<TreeNode*> tmpq;
            q.push(pRoot);
            int n = 1;

            while(!q.empty())
            {
                TreeNode* p = q.front();
                q.pop();
                tmp.push_back(p->val);
                --n;

                if(p->left)
                    tmpq.push(p->left);
                if(p->right)
                    tmpq.push(p->right);

                if(n == 0)
                {
                    ans.push_back(tmp);
                    tmp.clear();
                    while(!tmpq.empty())
                    {
                        TreeNode* p = tmpq.front();
                        tmpq.pop();
                        ++n;
                        q.push(p);
                    }
                }
            }
            return ans;
        }

};