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

题目描述

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

思路：

层次遍历，再用两个变量，一个记录当前层要打印的节点数，一个记录下一层要打印的节点数

C++

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
class Solution {
public:
        vector> Print(TreeNode* pRoot) {
            if(pRoot == nullptr)
                return  {};
            queue q;
            vector> res;
            q.push(pRoot);
            
            while(!q.empty()){
                int l = 0, r = q.size();
                vector temp;
                while(l++ < r){
                    TreeNode* pNode = q.front();
                    q.pop();
                    temp.push_back(pNode->val);
                    if(pNode->left != nullptr)
                        q.push(pNode->left);
                    if(pNode->right != nullptr)
                        q.push(pNode->right);
                }
                res.push_back(temp);
            }
            return res;
        }
    
};