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);
}
}
}
京公网安备 11010502036488号