把二叉树打印成多行

时间限制:C/C++ 1秒,其他语言2秒 空间限制:C/C++ 32M,其他语言64M

题目描述

从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。

思路:

广度优先搜索,借助队列

/*
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)
                return res;
            queue<TreeNode*> que;
            que.push(pRoot);
            while(!que.empty())
            {
                vector<int> vec;
                int len = que.size();
                for(int i =0;i < len;i++)
                {
                    TreeNode *temp = que.front();
                    vec.push_back(temp->val);
                    que.pop();
                    if(temp->left) que.push(temp->left);
                    if(temp->right) que.push(temp->right);
                }
                res.push_back(vec);
            }
            return res;
        }
    
};
posted @ 2020-04-13 17:50  牛犁heart  阅读(96)  评论(0编辑  收藏  举报