LintCode【简单】69. 二叉树的层次遍历。代码及思路

题目要求：

给出一棵二叉树，返回其节点值的层次遍历（逐层从左往右访问）

您在真实的面试中是否遇到过这个题？ 

Yes

样例

给一棵二叉树 

{3,9,20,#,#,15,7}

 ：

3
/ \
9  20
/  \
15   7

返回他的分层遍历结果：

[
[3],
[9,20],
[15,7]
]

思路：

层序遍历和前中后序遍历不一样，稍微复杂一些。思路注释在代码里了。

需要用到队列的知识，此处放一个链接：C++ 中queue（队列）的用法
4000

代码：

/**
* Definition of TreeNode:
* class TreeNode {
* public:
* int val;
* TreeNode *left, *right;
* TreeNode(int val) {
* this->val = val;
* this->left = this->right = NULL;
* }
* }
*/

class Solution {
public:
/*
* @param root: A Tree
* @return: Level order a list of lists of integer
*/
vector<vector<int>> levelOrder(TreeNode * root) {
// write your code here
vector<vector<int>> result;//建立一个二维的vector最后用来返回
if(root == NULL) return result;//如果是空，则直接返回

queue<TreeNode *> que;//建立一个存放树节点类型指针的队列que
que.push(root);//放入root节点

while(!que.empty()){//队列不是空（循环到队列为空结束）
int size = que.size();//队列中元素的个数，表示这一层有几个元素
vector<int> temp;
//定义一层的vector容器，接下来循环将队列里前size个元素放入这一层，出队列
for(int i = 0; i < size; i++){
TreeNode *q = que.front();//这里不定义指针，直接用que.front()->val也可以，下同
temp.push_back(q->val);
que.pop();
//判断有无左右孩子，放入队列
if(q->left) que.push(q->left);
if(q->right) que.push(q->right);
}
//循环结束后将这一层放入result
result.push_back(temp);
}
return result;
}
};