求:
给你一个二叉树,请你返回其按 层序遍历 得到的节点值。 (即逐层地,从左到右访问所有节点)。
示例:
二叉树:[3,9,20,null,null,15,7],
3
/
9 20
/
15 7
返回其层次遍历结果:
[
[3],
[9,20],
[15,7]
]
解:
思路:使用广度优先搜索,逐层扫描数据。这里需要一个辅助数据结构:队列。首先将根节点入队列,对每一层的扫描,都应该有一个数组保存结果。遍历当前层时,逐个从主队列中弹出数据,将数据加入到当前层的数组中,并将当前层下一层的数据放到一个临时队列中存储(这样才能知道下一层有哪些数据,并且由于队列FIFO的特性,能够保证下一次层序遍历时候的有序性)。当前层遍历完毕后,将临时队列中的数据加入到主队列中,继续进行下一层的迭代。当主队列为空时,说明所有数据已经遍历完毕,返回结果。
这里因为Java对队列、可变数组的快速实现,因此使用Java语言实现。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class
Solution {
public
List> levelOrder(TreeNode root) {
List> ret =
new
ArrayList();
if
(root==null)
return
ret;
Queue queue =
new
LinkedList<>();
queue.add(root);
while
(!queue.isEmpty()){
List levelList =
new
ArrayList();
Queue tmp =
new
LinkedList<>();
while
(!queue.isEmpty()) {
TreeNode t = queue.remove();
levelList.add(t.val);
if
(t.left != null) tmp.add(t.left);
if
(t.right != null) tmp.add(t.right);
}
ret.add(levelList);
queue = tmp;
}
return
ret;
}
}