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

从下到上打印二叉树

https://www.nowcoder.com/practice/ed982e032ff04d6a857b4cb4e6369d04

import java.util.*;

/*
 * public class TreeNode {
 *   int val = 0;
 *   TreeNode left = null;
 *   TreeNode right = null;
 *   public TreeNode(int val) {
 *     this.val = val;
 *   }
 * }
 */

public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     * 
     * @param root TreeNode类 
     * @return int整型二维数组
     */
    public ArrayList<ArrayList<TreeNode>> sol(TreeNode root){
        ArrayList<ArrayList<TreeNode>> res = new ArrayList<ArrayList<TreeNode>>();
        if(root == null) return res;
        Deque<TreeNode> deq = new ArrayDeque<TreeNode>();
        deq.add(root);
        while(!deq.isEmpty()){
            int len = deq.size();
            ArrayList<TreeNode> temp = new ArrayList<TreeNode>();
            for(int i = 0; i < len; i++){
                TreeNode node = deq.remove();
                temp.add(node);
                if(node.left!=null)deq.add(node.left);
                if(node.right!=null)deq.add(node.right);
            }
            res.add(temp);
        }
        return res;
    }


    public int[][] levelOrderBottom (TreeNode root) {
        // write code here
        ArrayList<ArrayList<TreeNode>> res = sol(root);
        Collections.reverse(res);
        int[][] ans = new int[res.size()][];
        for(int i = 0; i < res.size();i++){
            ArrayList<TreeNode> temp = res.get(i);
            int[] arr = new int[temp.size()];
            for(int j = 0; j < temp.size();j++){
                arr[j] = temp.get(j).val;
            }
            ans[i] = arr;
        }
        return ans;
    }
}

全部评论
面试抽到了这题,本来可以秒,我竟然不知道java二维数组是可以不设置第二维长度的,太可惜了
点赞 回复 分享
发布于 2023-04-10 17:27 北京

相关推荐

头像
03-20 22:00
重庆大学 Java
适彼乐土:“他们不行再找你” 最后的底牌吗?有点意思
点赞 评论 收藏
分享
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务