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

题目

给你一个二叉树的根结点 root ，请返回出现次数最多的子树元素和。如果有多个元素出现的次数相同，返回所有出现次数最多的子树元素和（不限顺序）。

一个结点的 「子树元素和」 定义为以该结点为根的二叉树上所有结点的元素之和（包括结点本身）。

示例

输入: root = [5,2,-3]
输出: [2,-3,4]

输入: root = [5,2,-5]
输出: [2]

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

方法1：递归

Java实现1

class Solution {
    List<Integer> res = new ArrayList<>();
    public int[] findFrequentTreeSum(TreeNode root) {
        if (root == null) return new int[]{};

        pd(root);
        
        Map<Integer, Integer> map = new HashMap<>();
        int flag = 1;
        for (int i = 0; i < res.size(); i++) {
            int a = res.get(i);
            if (!map.containsKey(a)) {
                map.put(a, 1);
            } else {
                map.put(a, map.get(a) + 1);
                if (map.get(a) > flag) flag = map.get(a);
            }
        }

        List<Integer> o1 = new ArrayList<>();
        for (int key : map.keySet()) {
            if (map.get(key) == flag) o1.add(key);
        }
        int[] out = new int[o1.size()];
        for (int i = 0; i < o1.size(); i++) out[i] = o1.get(i);

        return out;
    }

    public void pd(TreeNode root) {
        if (root == null) return;

        res.add(traverse(root));

        pd(root.left);
        pd(root.right);
    }

    public int traverse(TreeNode root){
        if (root == null) return 0;

        return root.val + traverse(root.left) + traverse(root.right);
    }
}

Java实现2

class Solution {
    List<Integer> res = new ArrayList<>();
    Map<Integer, Integer> map = new HashMap<>();
    int max = 0;
    public int[] findFrequentTreeSum(TreeNode root) {
        bl(root);
        int[] out = new int[res.size()];
        for (int i = 0; i < res.size(); i++) out[i] = res.get(i);
        return out;
    }

    public void bl(TreeNode root) {
        if (root == null) return;
        int var = traverse(root);
        map.put(var, map.getOrDefault(var, 0) + 1);
        if (map.get(var) > max) {
            res = new ArrayList<>();
            max = map.get(var);
            res.add(var);
        } else if (map.get(var) == max) {
            res.add(var);
        }

        bl(root.left);
        bl(root.right);
    }
    public int traverse(TreeNode root) {
        if (root == null) return 0;

        return root.val + traverse(root.left) + traverse(root.right);
    }
}