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

508. 出现次数最多的子树元素和https://leetcode.cn/problems/most-frequent-subtree-sum/

难度中等208

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

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

示例 1：

 

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

示例 2：

 

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

提示:

• 节点数在 [1, 104] 范围内
• -105 <= Node.val <= 105

通过次数40,599提交次数53,882

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    Map<Integer,Integer> map = new HashMap<Integer,Integer>();
    int max = 1;
    public int[] findFrequentTreeSum(TreeNode root) {
        dfs(root);
        int[] ans = new int[map.size()];
        int index =0;
        for (Integer i : map.keySet()) 
            if(map.get(i)==max)
                ans[index++]=i;
        int[] ans1 = Arrays.copyOf(ans,index);
        return ans1;
    }

    int dfs(TreeNode node)
    {
        if(node==null) return 0;
        int l = dfs(node.left);
        int r = dfs(node.right);
        int sum = node.val+l+r;
        if(map.containsKey(sum)==true)
        {
            map.put(sum,map.get(sum)+1);
            max = Math.max(max,map.get(sum));
        }
        else map.put(sum,1);
        return sum;
    }
}