通过万岁!!!
- 题目:给你一个二叉树,找他的子树和,子树和可能会重复,要求找到重复次数最多的那个,要是又多个重复次数一样的,就需要都返回,也就是返回一个数组。
- 思路:这个题其实有点绕,题目我最开是以为是子树相同,然后找他们的和,后来才发现是子树和相同。思路就是递归,每个递归函数找到当前这个数的和,然后放入map中,注意map的key是子树和,而value是出现的次数。最后遍历完这个树以后,我们就找到了所有节点作为根节点的和。然后我们遍历map找到value中最大值,并且找到value等于这个最大值的key,填充到数组中即可。只不过我在代码中直接找到最大值并且计算了最后这个数组的长度。
- 技巧:二叉树遍历、哈希
java代码
class Solution {
Map<Integer, Integer> map;// key是子树和,value是出现的次数
int max = 0;
int len = 0;
public int[] findFrequentTreeSum(TreeNode root) {
map = new HashMap<>();
getSonTreeSum(root);
int[] ans = new int[len];
int idx = 0;
for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
if (entry.getValue() == max) {
ans[idx++] = entry.getKey();
}
}
return ans;
}
private int getSonTreeSum(TreeNode root) {
if (root == null) {
return 0;
}
int sum = root.val;
sum += getSonTreeSum(root.left);
sum += getSonTreeSum(root.right);
map.putIfAbsent(sum, 0);
int i = map.get(sum) + 1;//
if (i > max) {
len = 1;
max = i;
} else if (i == max) {
++len;
}
map.put(sum, i);
return sum;
}
}
- 总结:题目不是很难,但是感觉题目本身有点语病,或者是我理解的问题吧,所以导致最开始思路出现了偏差。