【日常训练】508. 出现次数最多的子树元素和

最新推荐文章于 2025-07-03 13:57:59 发布

原创最新推荐文章于 2025-07-03 13:57:59 发布 · 214 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

收录于

leetcode

这篇博客讨论了一道关于二叉树的问题，即找到出现次数最多的子树元素和。通过深度优先搜索（DFS）遍历每个节点，计算子树元素和并更新到哈希表中记录出现次数。最后，找出出现次数最多的子树元素和并返回。示例展示了如何处理具有不同元素和的节点情况。

题目

给你一个二叉树的根结点 root ，请返回出现次数最多的子树元素和。如果有多个元素出现的次数相同，返回所有出现次数最多的子树元素和（不限顺序）。

一个结点的「子树元素和」定义为以该结点为根的二叉树上所有结点的元素之和（包括结点本身）。

示例 1：
在这里插入图片描述
输入: root = [5,2,-3]
输出: [2,-3,4]

示例 2：
在这里插入图片描述
输入: root = [5,2,-5]
输出: [2]

提示:
节点数在 [1, 10⁴] 范围内
-10⁵ <= Node.val <= 10⁵

代码

package dayLeetCode;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

public class dayleetcode508 {
    // dfs 对于每一颗子树，其子树元素和就是该节点的val加上左子树和右子树的val
    // 所以我们不妨将所有的子树元素和 和 它们出现的次数记录下来

    Map<Integer, Integer> map = new HashMap<>();
    // 用来记录出现次数最多的子树元素和出现了多少次
    int maxCnt = 0;

    public int[] findFrequentTreeSum(TreeNode root) {
        dfs(root);
        List<Integer> list = new ArrayList<>();

        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
            if (entry.getValue() == maxCnt){
                list.add(entry.getKey());
            }
        }

        int[] ans = new int[list.size()];
        int i = 0;
        for (int num : list){
            ans[i++] = num;
        }
        return ans;
    }

    private int dfs(TreeNode curNode) {
        if (curNode == null){
            return 0;
        }
        int sum = curNode.val + dfs(curNode.left) + dfs(curNode.right);
        map.put(sum, map.getOrDefault(sum,0) + 1);
        maxCnt = Math.max(maxCnt, map.get(sum));
        return sum;
    }


}