携手创作，共同成长！这是我参与「掘金日新计划 · 8 月更文挑战」的第5天，点击查看活动详情

题目详情

LeetCode题库序号 1161. 最大层内元素和 ，难度为 中等。

Tag : 「深度优先搜索」

给你一个二叉树的根节点 root。设根节点位于二叉树的第 1 层，而根节点的子节点位于第 2 层，依此类推。

请返回层内元素之和 最大 的那几层（可能只有一层）的层号，并返回其中 最小 的那个。

示例 1：

输入：root = [1,7,0,7,-8,null,null]
输出：2
解释：
第 1 层各元素之和为 1，
第 2 层各元素之和为 7 + 0 = 7，
第 3 层各元素之和为 7 + -8 = -1，
所以我们返回第 2 层的层号，它的层内元素之和最大。

示例 2：

输入： root = [989,null,10250,98693,-89388,null,null,null,-32127]
输出： 2

提示:

• 树中的节点数在 [1, 104]范围内
• -105 <= Node.val <= 105

深度优先搜索

题解思路：

这道题目是一道典型的DFS的解法的题目，因为从题目的意思，我们可以得知，我们需要求得最大的一层的和，那么我们就要对每一层进行求和，然后再比较每一层的和。最后得出拥有最大的和的那一层的层号。这道题目在最初的版本是可以使用Map进行作为临时存放的容器的。本次我们使用数组的作为存放的容器。数组的索引天然对应着我们所需要的层号。随后就是深度优先搜索的标准代码了。详情见以下代码：

题解代码

public class Solution {

    private List<Integer> ans = new ArrayList<>();

    public int maxLevelSum(TreeNode root) {
        dfs(root, 0);
        int res = 0;
        for (int i = 0; i < ans.size(); i++) {
            if (ans.get(i) > ans.get(res)) {
                res = i;
            }
        }
        return res + 1;
    }

    private void dfs(TreeNode node, Integer level) {
        if (level == ans.size()) {
            ans.add(node.val);
        } else {
            ans.set(level, node.val + ans.get(level));
        }

        if (node.left != null) {
            dfs(node.left, level + 1);
        }

        if (node.right != null) {
            dfs(node.right, level + 1);
        }
    }



    public class TreeNode {
        int val;
        TreeNode left;
        TreeNode right;

        TreeNode() {
        }

        TreeNode(int val) {
            this.val = val;
        }

        TreeNode(int val, TreeNode left, TreeNode right) {
            this.val = val;
            this.left = left;
            this.right = right;
        }
    }
}

结尾

我的"刷完LeetCode题库"系列文章的第 No.1161 序号的题目，本次刷题之旅系列开始于 2022-06-12，因为LeetCode上部分是有锁题，我自己的目标是将先把所有不带锁的题目刷完。自己能够通过这次刷题之旅勉励自己，并且提升逻辑思维能力。这个系列的文章就是会见证我自己的一个成长过程！

思路虽然不是最优的，但是我会尽我所能！

为了让我自己的刷题之旅不中断，我特地建立了相关的仓库，来记录我自己的刷题之旅。 github.com/jackpan123/… 。