题目描述:
从上到下按层打印二叉树,同一层的节点按从左到右的顺序打印,每一层打印到一行。
例如:
给定二叉树: [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
返回其层次遍历结果:
[
[3],
[9,20],
[15,7]
]
题目解析:
由于需要分层打印,我们类似不分层时候的操作,仍旧是建立队列存储节点,但是在每层与层之间加入一个NULL作为分隔,每次遇到NULL的时候就开始下一层的打印。
class Solution {
public:
vector<vector<int>> levelOrder(TreeNode* root) {
queue<TreeNode*> q;
q.push(root);
vector<vector<int>> res={};
if(root==NULL)
return res;
while(!q.empty())
{
q.push(NULL);
vector<int> t;
while(q.front()!=NULL)
{
if(q.front()->left!=NULL)
q.push(q.front()->left);
if(q.front()->right!=NULL)
q.push(q.front()->right);
t.push_back(q.front()->val);
q.pop();
}
q.pop();
res.push_back(t);
}
return res;
}
};