其他分享
首页 > 其他分享> > [LC515]在每个树的行中找最大值

[LC515]在每个树的行中找最大值

作者:互联网

题目内容

题目分析

这是一道典型的树结构遍历题,可以使用层序遍历(BFS)或者(DFS)进行解题。

BFS
    public List<Integer> largestValues(TreeNode root) {
        Deque<TreeNode> queue = new ArrayDeque();
        if (root != null) queue.add(root);
        List<Integer> res = new ArrayList();
        while(!queue.isEmpty()){
            int sz = queue.size();
            Integer levelMax = Integer.MIN_VALUE;
            for(int i = 0; i < sz; i++){
                TreeNode node = queue.pollFirst();
                levelMax = Math.max(levelMax, node.val);
                if (node.left != null) queue.addLast(node.left);
                if (node.right != null) queue.addLast(node.right);
            }
            res.add(levelMax);
        }
        return res;
    }
DFS
    int maxDepth = 0;
    Map<Integer, Integer> map = new HashMap<>();
    public List<Integer> largestValues(TreeNode root) {
        List<Integer> res = new ArrayList<>();
        dfs(root, 1);
        for (int i = 1; i <= max; i++) res.add(map.get(i));
        return res;
    }
    void dfs(TreeNode node, int depth) {
        if (node == null) return ;
        maxDepth = Math.max(maxDepth, depth);
        map.put(depth, Math.max(map.getOrDefault(depth, Integer.MIN_VALUE), node.val));
        dfs(node.left, depth + 1);
        dfs(node.right, depth + 1);
    }

标签:node,LC515,int,res,最大值,行中,queue,new,root
来源: https://www.cnblogs.com/xy1997/p/16410447.html