题解 | #从上往下打印二叉树#

从上往下打印二叉树

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

思路: 用递归来表示。一层一层来,层数跟二维数组行数对应。递归时候记得深度加一。

全部评论

相关推荐

点赞 评论 收藏
分享
01-15 13:52
已编辑
河南大学 Java
六年要多久:标准头像,不吃香菜😂
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务