leetcode 508. 出现次数最多的子树元素和

给你一个二叉树的根结点，请你找出出现次数最多的子树元素和。一个结点的「子树元素和」定义为以该结点为根的二叉树上所有结点的元素之和（包括结点本身）。

你需要返回出现次数最多的子树元素和。如果有多个元素出现的次数相同，返回所有出现次数最多的子树元素和（不限顺序）。

 

示例 1：
输入:

5
/ \
2 -3
返回 [2, -3, 4]，所有的值均只出现一次，以任意顺序返回所有值。

示例 2：
输入：

5
/ \
2 -5
返回 [2]，只有 2 出现两次，-5 只出现 1 次。

 

提示： 假设任意子树元素和均可以用 32 位有符号整数表示。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/most-frequent-subtree-sum
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

后序遍历，用map记录本节点和左右节点的和，并返回，最后遍历map，找出最大value的key.

 public int[] findFrequentTreeSum(TreeNode root) {
        Map<Integer, Integer> map = new HashMap<>();
        find(root, map);
        int max = 0;
        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
            max = Math.max(max, entry.getValue());
        }
        List<Integer> collect = new ArrayList<>();

        for (Map.Entry<Integer, Integer> entry : map.entrySet()) {
            if (entry.getValue() == max) {
                collect.add(entry.getKey());
            }
        }
        int size = collect.size();
        int[] arr = new int[size];
        for (int i = 0; i < size; i++) {
            arr[i] = collect.get(i);
        }
        return arr;
    }

    private int find(TreeNode root, Map<Integer, Integer> map) {
        if (root == null) {
            return 0;
        }
        int l = find(root.left, map);
        int r = find(root.right, map);
        int value = root.val + l + r;
        map.merge(value, 1, (a, b) -> a + b);
        return value;
    }