题目描述
从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
解题思路
直接采用层次遍历法。
/*
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> > res;
if(pRoot == NULL) return res;
queue<TreeNode*> que; //建立结点指针队列
que.push(pRoot); //将头结点指针存入队列
while(!que.empty()){
int len = que.size(); //求出当前层的数量len,之后读len个结点指针
vector< int > tmp;
while(len--){
TreeNode* p = que.front(); //读出队列首部指针
que.pop(); //删除
tmp.push_back(p->val);
if(p->left) que.push(p->left); //将它存在的左右子结点存入队列末尾
if(p->right) que.push(p->right);
}
res.push_back(tmp);
}
return res;
}
};