题目描述
从上到下按层打印二叉树,同一层结点从左至右输出。每一层输出一行。
解题思路:
二叉树层次遍历, 利用一个队列,每层之间用一个NULL分割开,当遇到NULL则代表当前层已遍历完
AC C++ Solution:
class Solution {
public:
vector<vector<int> > Print(TreeNode* pRoot) {
vector<vector<int> > res;
if(!pRoot) return res;
queue<TreeNode*> q;
q.push(pRoot);
q.push(NULL);
vector<int> cur_vec;
while(!q.empty()) {
TreeNode *t = q.front();
q.pop();
if(t == NULL) {
res.push_back(cur_vec);
cur_vec.resize(0);
if(q.size() > 0)
q.push(NULL); //每层之间用一个NULL作为分隔符
}
else { //把每个节点的值存入cur_vec,并把下一层的节点加入队列
cur_vec.push_back(t->val);
if(t->left) q.push(t->left);
if(t->right) q.push(t->right);
}
}
return res;
}
};