递归求解二叉树层序遍历(Java)
求二叉树的层序遍历
http://www.nowcoder.com/questionTerminal/04a5560e43e24e9db4595865dc9c63a3
Java递归,直接从某力的二叉树卡片过来,
思想很简单,首先声明一个成员变量供全局使用;
其次写一个向下探索的递归方法,注意先从左边再从右边;每向下一层level+1
这里很巧妙的是,对于二维List来说,将根视为第0层,树的深度正好等于一维List的个数。
import java.util.*;
/*
* public class TreeNode {
* int val = 0;
* TreeNode left = null;
* TreeNode right = null;
* }
*/
public class Solution {
/**
*
* @param root TreeNode类
* @return int整型ArrayList<ArrayList<>>
*/
ArrayList<ArrayList<Integer>> res = new ArrayList<ArrayList<Integer>>();
public ArrayList<ArrayList<Integer>> levelOrder (TreeNode root) {
// write code here
if(root == null){
return res;
}
count(root,0);
return res;
}
public void count(TreeNode node, int level){
if(level == res.size()){
res.add(new ArrayList<Integer>());
}
ArrayList<Integer> list = res.get(level);
list.add(node.val);
if(node.left != null){
count(node.left, level+1);
}
if(node.right != null){
count(node.right, level+1);
}
}
}
文远知行公司福利 495人发布
查看19道真题和解析