题目要求
给定一个二叉树,返回节点值的层序遍历结果。
(tips:从左到右,一层一层)
输入示例
For example:
给定一个二叉树 [3,9,20,null,null,15,7],
3
/ \
9 20
/ \
15 7
return 层序遍历结果如下:
[
[3],
[9,20],
[15,7]
]
解题思路
根据输入示例的结果形式可知,我们需要两个vector来辅助解答。
一个vector存放每一层的遍历结果,一个vector存放最后的整体结果。
根据层序遍历的要求(从左到右,一层一层的特点),我们选用队列来实现(先进先出),试想一下,每次我们取队首元素,然后把它的值放入当前层的vector中,再把它的孩子节点放到队列里进行迭代,只要队列中还有元素我们就进行循环操作,直到到达叶子结点。
, 难点是:什么时候我们结束当前一层遍历去进行下一层呢?
答案:这里通过当前层中-队列-里元素的个数就能知道当前层进行几次循环取值操作,用for来进行循环取值。比如第二层有9,20两个元素,要进行两个for循环(注意这里有个大坑点,不要在for循环中使用q.size()因为这个队列q在for循环中是不断改变的)
主要代码(c++)
/**
* 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) {
if(root==NULL) return {};
vector<vector<int>>res; // 第一个vector
queue<TreeNode*>q;
q.push(root);
while(!q.empty())// 迭代的条件
{
int size = q.size(); //这里要先保存下来队列的元素个数,不然在for中会改变
vector<int>currentlevel; //第二个vector
for(int i =0; i<size; ++i) //大坑点! 你可以把size换成q.size()试试看
{
TreeNode * tmp = q.front(); // 下面是常规写法 记录每一层的值
q.pop();
currentlevel.push_back(tmp->val);
if(tmp->left !=NULL) q.push(tmp->left);
if(tmp->right !=NULL) q.push(tmp->right);
}
res.push_back(currentlevel);
}
return res;
}
};
主要代码python
# Definition for a binary tree node.
# class TreeNode(object):
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution(object):
def levelOrder(self, root):
"""
:type root: TreeNode
:rtype: List[List[int]]
"""
queue = []
res = []
if root is None:
return res
queue.append(root)
while len(queue)!=0:
eachlevel = []
size = len(queue)
for i in range(size):
temp = queue.pop(0)
eachlevel.append(temp.val)
if temp.left:
queue.append(temp.left)
if temp.right:
queue.append(temp.right)
res.append(eachlevel)
return res
划重点:
有关层序遍历的迭代记得用 队列哈!!!
相似题目:
解答:leetcode107 Binary Tree Level Order Traversal 2 (二叉树的层序遍历2)