题目
Given a binary tree, return the level order traversal of its nodes' values. (ie, from left to right, level by level).
For example:
Given binary tree {3,9,20,#,#,15,7}
,
3
/ \
9 20
/ \
15 7
return its level order traversal as:
[
[3],
[9,20],
[15,7]
]
解答
1.递归法:根据外层的List的长度size和层数level的对应的关系使用递归(详细见代码注释),代码如下:
/**
* Definition for binary tree
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
public class Solution {
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> ret=new ArrayList<List<Integer>>();
dfs(root,0,ret);
return ret;
}
//分层遍历-递归
private static void dfs(TreeNode root,int level,List<List<Integer>> ret){
if(root==null){
return;
}
//添加一个新的ArrayL表示新的一层
if(level>=ret.size()){
ret.add(new ArrayList<Integer>());
}
ret.get(level).add(root.val); //把节点添加到表示那一层的ArrayList里
dfs(root.left,level+1,ret); //递归处理下一层的左子树和右子树
dfs(root.right,level+1,ret);
}
}
2.迭代法:相当于广度优先搜索,使用队列实现。队列初始化,将根节点压入队列,当队列不为空,进行如下操作:弹出一个节点,并访问其左右子节点,若左子结点或右子节点不为空,将其压入队列,同时记录当前层节点访问变化和下一层的节点数。代码如下:
public List<List<Integer>> levelOrder(TreeNode root) {
List<List<Integer>> list=new ArrayList<List<Integer>>();
if(root==null){
return list;
}
Queue<TreeNode> queue=new LinkedList<TreeNode>();
queue.add(root);
List<Integer> item=new ArrayList<Integer>();
int currentLevel=1;
int nextLevel=0;
while(!queue.isEmpty()){
TreeNode cur=queue.remove();
currentLevel--;
item.add(cur.val);
if(cur.left!=null){
queue.add(cur.left);
nextLevel++;
}
if(cur.right!=null){
queue.add(cur.right);
nextLevel++;
}
if(currentLevel==0){
list.add(item);
item=new ArrayList<Integer>();
currentLevel=nextLevel;
nextLevel=0;
}
}
return list;
}
---EOF---