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

题目描述

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

 

解题思路：可以直到，每次打印一个结点，然后打印下一行时总是先打印其左子结点，然后打印其右子结点。先入先出，使用队列处理。使用toBeprint记录当前行打印的个数，每存入一个-1；nextlevel记录下一行打印结点个数；一行打印完，toBeprint = 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> > result;
            vector<int> temp;//间接变量，存储一行
            if(pRoot == NULL)
                return result;
            queue<TreeNode *> nodes;
            nodes.push(pRoot);
            int nextlevel = 0;//下一行应打印个数
            int toBePrint = 1;//当前行剩余打印个数
            while(!nodes.empty())
            {
                TreeNode *pNode = nodes.front();
                temp.push_back(pNode->val);
                if(pNode->left != NULL)
                {
                    nodes.push(pNode->left);
                    ++nextlevel;
                }
                if(pNode->right != NULL)
                {
                    nodes.push(pNode->right);
                    ++nextlevel;
                }
                nodes.pop();
                --toBePrint;
                if(toBePrint == 0)//一行已打印完
                {
                    result.push_back(temp);
                    temp.clear();//清空
                    toBePrint = nextlevel;
                    nextlevel = 0;
                }
            }
            return result;
        }
};