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); } } }