|
| 1 | +### 题目描述 |
| 2 | + |
| 3 | +这是 LeetCode 上的 **[508. 出现次数最多的子树元素和](https://leetcode.cn/problems/most-frequent-subtree-sum/solution/by-ac_oier-t3y4/)** ,难度为 **中等**。 |
| 4 | + |
| 5 | +Tag : 「树的遍历」、「DFS」、「哈希表」 |
| 6 | + |
| 7 | + |
| 8 | + |
| 9 | +给你一个二叉树的根结点 `root`,请返回出现次数最多的子树元素和。如果有多个元素出现的次数相同,返回所有出现次数最多的子树元素和(不限顺序)。 |
| 10 | + |
| 11 | +一个结点的 「子树元素和」 定义为以该结点为根的二叉树上所有结点的元素之和(包括结点本身)。 |
| 12 | + |
| 13 | +示例 1: |
| 14 | + |
| 15 | +``` |
| 16 | +输入: root = [5,2,-3] |
| 17 | +
|
| 18 | +输出: [2,-3,4] |
| 19 | +``` |
| 20 | +示例 2: |
| 21 | + |
| 22 | +``` |
| 23 | +输入: root = [5,2,-5] |
| 24 | +
|
| 25 | +输出: [2] |
| 26 | +``` |
| 27 | + |
| 28 | +提示: |
| 29 | +* 节点数在 $[1, 10^4]$ 范围内 |
| 30 | +* $-10^5 <= Node.val <= 10^5$ |
| 31 | + |
| 32 | +--- |
| 33 | + |
| 34 | +### DFS |
| 35 | + |
| 36 | +使用 `DFS` 对树进行遍历,同时将每棵子树的总和值存入哈希表(以元素和为 `key`,出现次数为 `value`),并实时维护最大子树和,使用变量 `max` 存储。 |
| 37 | + |
| 38 | +当遍历完整棵树后,将哈希表中值为 `max` 的键存入答案。 |
| 39 | + |
| 40 | +代码: |
| 41 | +```Java |
| 42 | +class Solution { |
| 43 | + Map<Integer, Integer> map = new HashMap<>(); |
| 44 | + int max = 0; |
| 45 | + public int[] findFrequentTreeSum(TreeNode root) { |
| 46 | + dfs(root); |
| 47 | + List<Integer> list = new ArrayList<>(); |
| 48 | + for (int k : map.keySet()) { |
| 49 | + if (map.get(k) == max) list.add(k); |
| 50 | + } |
| 51 | + int n = list.size(); |
| 52 | + int[] ans = new int[n]; |
| 53 | + for (int i = 0; i < n; i++) ans[i] = list.get(i); |
| 54 | + return ans; |
| 55 | + } |
| 56 | + int dfs(TreeNode root) { |
| 57 | + if (root == null) return 0; |
| 58 | + int cur = root.val + dfs(root.left) + dfs(root.right); |
| 59 | + map.put(cur, map.getOrDefault(cur, 0) + 1); |
| 60 | + max = Math.max(max, map.get(cur)); |
| 61 | + return cur; |
| 62 | + } |
| 63 | +} |
| 64 | +``` |
| 65 | +* 时间复杂度:$O(n)$ |
| 66 | +* 空间复杂度:$O(n)$ |
| 67 | + |
| 68 | +--- |
| 69 | + |
| 70 | +### 最后 |
| 71 | + |
| 72 | +这是我们「刷穿 LeetCode」系列文章的第 `No.508` 篇,系列开始于 2021/01/01,截止于起始日 LeetCode 上共有 1916 道题目,部分是有锁题,我们将先把所有不带锁的题目刷完。 |
| 73 | + |
| 74 | +在这个系列文章里面,除了讲解解题思路以外,还会尽可能给出最为简洁的代码。如果涉及通解还会相应的代码模板。 |
| 75 | + |
| 76 | +为了方便各位同学能够电脑上进行调试和提交代码,我建立了相关的仓库:https://github.com/SharingSource/LogicStack-LeetCode 。 |
| 77 | + |
| 78 | +在仓库地址里,你可以看到系列文章的题解链接、系列文章的相应代码、LeetCode 原题链接和其他优选题解。 |
| 79 | + |
0 commit comments