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

题目描述

 

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

 

解题思路

还是用层序遍历的思想，设置一个队列初始推入根节点，每次删除队列顶的节点时，打印出来结点值，若其有左右子节点，则将左右子节点依次推入队列，如此遍历直到队列为空。

由于本题要求逐行输出，所以需要设置两个变量来保持分行：needPrint记录当前行仍需打印的节点数，nextLevel记录下一行需要打印的节点数。每打印完一个节点，将当前行剩下的结点数needPrint减1；每向队列放入一个子节点，将下一行的节点数nextLevel加1。打印完节点后若needPrint为0，则说明当前行节点已全部打印完，所以将needPrint设置为下一行的节点数nextLevel，同时将nextLevel置0.

代码

/*
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> > vs;
            if(pRoot){
                vector<int> v;
                queue<TreeNode*> q;
                q.push(pRoot);
                int needPrint=1;
                int nextLevel=0;
                while(q.size()){
                    TreeNode* node=q.front();
                    v.push_back(node->val);
                    needPrint--;
                    q.pop();
                    if(node->left){
                        q.push(node->left);
                        nextLevel++;
                    }
                    if(node->right){
                        q.push(node->right);
                        nextLevel++;
                    }
                    if(needPrint==0){
                        vs.push_back(v);
                        v.clear();
                        needPrint=nextLevel;
                        nextLevel=0;
                    }
                }
            }
            return vs;
        }
    
};