剑指offer--把二叉树打印成多行(C++)

题目描述:从上到下按层打印二叉树,同一层节点从左往右输出,每一层输出一行。
问题分析:这个问题就是二叉树的层序遍历,之前我的博客中曾写到过“锯齿型层序打印二叉树”,和本体思路基本一样,并且本题相对那个题较简单一些,我将那篇博客的连接放在这里:https://blog.csdn.net/cherrydreamsover/article/details/81051663
代码实现:
/*
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>> res;
            if(pRoot==NULL)
                return res;

            queue<TreeNode*> q;
            q.push(pRoot);

            while(!q.empty())
            {
                //用n来保存每一次d队列的元素个数,即就是二叉树每一层节点的个数
                int n=q.size();
                vector<int> v;
                while(v.size() < n)
                {
                    TreeNode* pRoot=q.front();
                    q.pop();
                    v.push_back(pRoot->val);

                    if(pRoot->left)
                    {
                        q.push(pRoot->left);
                    }
                    if(pRoot->right)
                    {
                        q.push(pRoot->right);
                    }
                }
                res.push_back(v);
            }

            return res;
        }

};

猜你喜欢

转载自blog.csdn.net/cherrydreamsover/article/details/81515076