题目:从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
因为要按层打印,首先想到的是层次遍历。在二叉树的深度这道题中,首先应用到了层次遍历。每一层的节点值存入一个小vector c,再把小vector c存到大vector vec中,打印vec。(题目没有要求换行,所以不需要考虑换行符)
class Solution {
public:
vector<vector<int> > Print(TreeNode* pRoot) {
vector<vector<int>> vec;
if(pRoot == NULL)
return vec;
queue<TreeNode*> que;
que.push(pRoot);
while(!que.empty()){
int depth = 0;
int len = que.size();
vector<int> c;
while(depth<len){
depth++;
TreeNode* node = que.front();
que.pop();
c.push_back(node->val);
if(node->left)
que.push(node->left);
if(node->right)
que.push(node->right);
}
vec.push_back(c);
}
return vec;
}
};