一、题目
给定一个二叉树,返回其按层次遍历的节点值。 (即逐层地,从左到右访问所有节点)。
例如:
给定二叉树: [3,9,20,null,null,15,7],
返回其层次遍历的结果:
二、思路——BFS(迭代)
利用队列实现,将树中所有的节点按层次依次入队、出队,从而实现层次遍历。过程如下:
- 创建一个列表,用以存储遍历的结果;
- 创建一个队列,刚开始先把根节点入队;
- 只要队列不为空,依次出队队列中所有的节点,并将节点的值加入列表中,再将所有节点的孩子入队。如此,一次循环完毕,树中一层的节点也就处理完毕,进入下一层循环;
- 队列为空时,遍历完毕,返回列表。
三、代码
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> list = new ArrayList<List<Integer>>();
if(root == null)
return list;
LinkedList<TreeNode> queue = new LinkedList<>();
queue.add(root);
while(!queue.isEmpty()){
List<Integer> list1 = new ArrayList<>();
TreeNode temp = null;
int size = queue.size();
for(int i = 0; i < size; i++){
temp = queue.poll();
list1.add(temp.val);
if(temp.left != null)
queue.add(temp.left);
if(temp.right != null)
queue.add(temp.right);
}
list.add(list1);
}
return list;
}
}