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

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

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

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

示例 1：
输入:

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

示例 2：
输入：

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

题解：
递归求每个子树的和，然后将子树的和作为key,子树和出现的个数作为value,存入到map中，再从map中找到value最大的值即可，
java代码：

  static HashMap<Integer, Integer> map = new HashMap<>();
    static int max = 0;

    /**
     * 出现次数最多的子树元素和
     *
     * @param root
     * @return
     */
    public static int[] findFrequentTreeSum(TreeNode root) {
        if (root == null) return new int[0];
        findSum(root);
        ArrayList<Integer> list = new ArrayList<>();
        for (int key : map.keySet()) {
            if (max == map.get(key)) {
                list.add(key);
            }
        }
        int[] res = new int[list.size()];
        for (int i = 0; i < list.size(); i++) {
            res[i] = list.get(i);
        }
        return res;
    }


    /**
     * @param root
     * @return
     */
    public static int findSum(TreeNode root) {
        if (root == null) return 0;
        int left = findSum(root.left);
        int right = findSum(root.right);
        int sum = root.value + left + right;
        //将结果放入HashMap中，并计算出出现的最多次数
        map.put(sum, map.getOrDefault(sum, 0) + 1);
        max = Math.max(max, map.get(sum));
        return sum;
    }