给你一个二叉树的根结点 root ,请返回出现次数最多的子树元素和。如果有多个元素出现的次数相同,返回所有出现次数最多的子树元素和(不限顺序)。
一个结点的 「子树元素和」 定义为以该结点为根的二叉树上所有结点的元素之和(包括结点本身)。
示例 1:
输入: root = [5,2,-3]
输出: [2,-3,4]
示例 2:
输入: root = [5,2,-5]
输出: [2]
深度优先搜索算法
class Solution {
int max = 0;
public int[] findFrequentTreeSum(TreeNode root) {
Map<Integer, Integer> map = new HashMap<>();
helper(root, map);
List<Integer> list = new ArrayList<>();
for (Integer i : map.keySet()) {
if (map.get(i) == max) {
list.add(i);
}
}
return list.stream().mapToInt(Integer::valueOf).toArray();
}
int helper(TreeNode root, Map<Integer, Integer> map) {
if (root == null) {
return 0;
}
int left = helper(root.left, map);
int right = helper(root.right, map);
int val = left + right + root.val;
map.put(val, map.getOrDefault(val, 0) + 1);
max = Math.max(max, map.get(val));
return val;
}
}
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/most-frequent-subtree-sum
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。