Given a binary tree, return the level order traversal of its nodes' values. (ie, from left to right, level by level).
For example:
Given binary tree [3,9,20,null,null,15,7]
,
3 / \ 9 20 / \ 15 7
return its level order traversal as:
[ [3], [9,20], [15,7] ]
思路分析:
层序遍历二叉树的常规思路:
先将树的根节点入队,
如果队列不空,则进入循环
{
将队首元素出队,并输出它;
如果该队首元素有左孩子,则将其左孩子入队;
如果该队首元素有右孩子,则将其右孩子入队;
}
该题目中要求返回按层返回遍历结果。
我们可以增加一个循环,每次处理一层的结点(从队列中弹出该层的所有节点,并将该层结点的孩子结点们入队列)。
代码:
/**
* Definition for a binary tree node.
* struct TreeNode {
* int val;
* TreeNode *left;
* TreeNode *right;
* TreeNode(int x) : val(x), left(NULL), right(NULL) {}
* };
*/
class Solution {
public:
vector< vector<int> > levelOrder(TreeNode* root) {
//二叉树的每一层用一个vector存储
vector< vector<int> > res;
//空树
if(!root) return res;
queue<TreeNode *> q;
q.push(root);
while(!q.empty()){
vector<int> everylevel;//记录每一层的结点值
int size = q.size();//同一层的结点个数
for(int i = 0; i < size; i ++){//处理某一层的所有节点:将该层的结点从队列中取出,并将下一层的结点(该层结点的孩子们入队列)
TreeNode *t = q.front();
//t出队列
q.pop();
everylevel.push_back(t -> val);
//将t的左右孩子入队列
if(t -> left)
q.push(t -> left);
if(t -> right)
q.push(t -> right);
}
res.push_back(everylevel);
}
return res;
}
};