题解 | #从上往下打印二叉树#
从上往下打印二叉树
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);
}
}
思路: 用递归来表示。一层一层来,层数跟二维数组行数对应。递归时候记得深度加一。




