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

题目描述

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

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

示例 1：

 

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

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

算法思想

        这题最容易想到的解法就是逐个元素求子树元素和，并将求得的结果作为key放入字典中，最后从values中找到一个出现次数最多的key，再遍历字典，把value=max_count的key存入ans中返回，就可以得到出现次数最多，可能有相同出现次数的一个子树元素和列表。

        不过时间复杂度和空间复杂度都不是很理想，这种暴力求解的方式虽然直观，但效率太低。

代码实现

class Solution:
    def findFrequentTreeSum(self, root: TreeNode) -> List[int]:
        dic={}
        ls=[]
        ans=[]
        max_count=0
        def getSum(root:TreeNode)->int:
            s=0
            if root:
                s+=root.val
            if root.left:
                s+=getSum(root.left)
            if root.right:
                s+=getSum(root.right)
            return s
        
        ls.append(root)
        while ls:
            root=ls.pop()
            s=getSum(root)
            dic[s]=dic.get(s,0)+1
            if root.left:
                ls.append(root.left)
            if root.right:
                ls.append(root.right)
        for value in dic.values():
            if value>max_count:
                max_count=value
        for key in dic.keys():
            if dic[key]==max_count:
                ans.append(key)
        return ans