题解 | #从上往下打印二叉树#
从上往下打印二叉树
https://www.nowcoder.com/practice/7fe2212963db4790b57431d9ed259701
import java.util.ArrayList; /** public class TreeNode { int val = 0; TreeNode left = null; TreeNode right = null; public TreeNode(int val) { this.val = val; } } */ public class Solution { public ArrayList<Integer> PrintFromTopToBottom(TreeNode root) { ArrayList<Integer> res = new ArrayList<>(); // 如果用递归的话,可以用二维数组。把树的深度和行数对应,最后把二维数组转化为一维数组即可 ArrayList<ArrayList<Integer>> list = new ArrayList<>(); transverse(root,list,1); for(int i = 0; i < list.size(); i++){ ArrayList<Integer> row = list.get(i); for(int j = 0; j < row.size(); j++){ res.add(row.get(j)); } } return res; } private void transverse( TreeNode root , ArrayList<ArrayList<Integer>> list , int depth){ if( root == null ){ return ; } // 说明需要增加新的一行,深度大于现有行数 if( list.size() < depth){ ArrayList<Integer> row = new ArrayList<>(); row.add(root.val); list.add(row); }else{ // 不需要添加行,在现有行里面进行 ArrayList<Integer> row = list.get(depth-1); row.add(root.val); } transverse(root.left,list,depth+1); transverse(root.right,list,depth+1); } }
思路: 用递归来表示。一层一层来,层数跟二维数组行数对应。递归时候记得深度加一。