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

题目描述：

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

本题知识点：

队列  树

解题思路：

层次遍历，每次记录当前层的结点个数，遍历完一层后再重新开始。

代码：

/*
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;
        if(!pRoot)
            return result;
        queue<TreeNode*> temp;
        temp.push(pRoot);
        while(!temp.empty())
        {
            int i = 0;
            int depth = temp.size();
            vector<int> cur;
            while(i++ < depth)
            {
                TreeNode* node = temp.front();
                temp.pop();
                cur.push_back(node->val);
                if(node->left)
                    temp.push(node->left);
                if(node->right)
                    temp.push(node->right);
            }
            result.push_back(cur);
        }
        return result;
    }
};