下午翻看了下leetcode,看了下今天的每日一题,主要还是考察二叉树的后续遍历。
题目链接:508. 出现次数最多的子树元素和
给你一个二叉树的根结点 root ,请返回出现次数最多的子树元素和。如果有多个元素出现的次数相同,返回所有出现次数最多的子树元素和(不限顺序)。
一个结点的 「子树元素和」 定义为以该结点为根的二叉树上所有结点的元素之和(包括结点本身)。
Map<Integer, Integer> map = new HashMap();
int max = 0;
public int[] findFrequentTreeSum(TreeNode root) {
findTreeSum(root);
//这里用 Iterator 迭代性能会差点
List<Integer> lst = new ArrayList<>();
for (Integer key : map.keySet()) {
if (map.get(key) == max) {
lst.add(key);
}
}
int[] arr = new int[lst.size()];
for (int i = 0; i < arr.length; i++) {
arr[i] = lst.get(i);
}
return arr;
}
public int findTreeSum(TreeNode root) {
if (root == null) {
return 0;
}
int left = findTreeSum(root.left);
int right = findTreeSum(root.right);
int sum = root.val + left + right;
map.put(sum, map.getOrDefault(sum,0)+1);
max = Math.max(max,map.get(sum));
return sum;
}
后续遍历的模板代码如下;
public int findTreeSum(TreeNode root) {
if (root == null) {
return 0;
}
int left = findTreeSum(root.left);
int right = findTreeSum(root.right);
//、、、
}