给定一个二叉树,返回其按层次遍历的节点值。 (即逐层地,从左到右访问所有节点)。
例如:
给定二叉树: [3,9,20,null,null,15,7],
3
/
9 20
/
15 7
返回其层次遍历结果:
[
[3],
[9,20],
[15,7]
]
层次遍历思想:
不同于前中后序遍历,是使用DFS,而层次遍历使用的思想是BFS(广度优先),具体实现方法为:
- (1)用队列保存当前节点的值(前中后序使用栈或递归)
- (2)如果当前节点值左孩子,不为空,就加入队列,如果当前节点值右孩子不为空,则计入队列,知道节点为空时停止。
但是本题返回List<List>,所以需要对基本的层次遍历进行一定方法的变化,但基本思想不变。List中装有一个list
- 具体到本题中,使用队列来保存从树中取出的节点,泛型当然为<TreeNode>
- 使用count来记录队列中的节点数目(count为每一层中节点数目)
- 新建一个TreeNode节点来保存从队列中弹出的节点,从而取出节点值,并判断当前节点是否有左右孩子,进而向下一层次进行遍历。
/**
* 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>> res = new LinkedList<List<Integer>>();
if(root == null)
return res;
Queue<TreeNode> q = new LinkedList<TreeNode>();
q.add(root);
while(!q.isEmpty())
{
List<Integer> list = new ArrayList<Integer>();
int count = q.size();
while(count>0)
{
TreeNode tmp = q.poll();
list.add(tmp.val);
if(tmp.left != null) q.add(tmp.left);
if(tmp.right != null) q.add(tmp.right);
count--;
}
res.add(list);
}
return res;
}
}