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

题目描述：

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

思路：

层次遍历，使用队列先进先出
设两个变量toBeAdd，nextLevel分别表示当前行需要遍历的节点数和下一层的节点数。

代码：

语言C++，已通过牛客网在线测试。

/*
struct TreeNode {
    int val;
    struct TreeNode *left;
    struct TreeNode *right;
    TreeNode(int x) :
            val(x), left(NULL), right(NULL) {
    }
};
*/
//层次遍历，使用队列先进先出
//设两个变量toBeAdd nextLevel分别表示当前行需要遍历的节点数和下一层的节点数
class Solution {
public:
        vector<vector<int> > Print(TreeNode* pRoot) {
            vector<vector<int>> res;
            if(pRoot == NULL) return res;
            queue<TreeNode *> que;
            que.push(pRoot);
            int toBeAdd = 1, nextLevel = 0;
            vector<int> tmp;
            while(!que.empty()){
                TreeNode* node = que.front();
                tmp.push_back(node->val);
                que.pop();
                toBeAdd--;
                if(node->left){
                    que.push(node->left);
                    nextLevel++;
                }
                if(node->right){
                    que.push(node->right);
                    nextLevel++;
                }
                if(toBeAdd == 0){
                    res.push_back(tmp);
                    tmp.clear();
                    toBeAdd = nextLevel;
                    nextLevel = 0;
                }
            }
            return res;
        }
};