Python描述 LeetCode 508. 出現次數最多的子樹元素和

大家好，我是亓官劼（qí guān jié ），在【亓官劼】公眾號、CSDN、GitHub、B站等平台分享一些技術博文，主要包括前端開發、python後端開發、小程序開發、數據結構與算法、docker、Linux常用運維、NLP等相關技術博文，時光荏苒，未來可期，加油~

如果喜歡博主的文章可以關注博主的個人公眾號【亓官劼】(qí guān jié)，裡面的文章更全更新更快。如果有需要找博主的話可以在公眾號後台留言，我會盡快回復消息.

---

本文原創為【亓官劼】（qí guān jié ），請大家支持原創，部分平台一直在惡意盜取博主的文章！！！ 全部文章請關注微信公眾號【亓官劼】。

題目

給你一個二叉樹的根結點 root ，請返回出現次數最多的子樹元素和。如果有多個元素出現的次數相同，返回所有出現次數最多的子樹元素和（不限順序）。

一個結點的 「子樹元素和」 定義為以該結點為根的二叉樹上所有結點的元素之和（包括結點本身）。

示例 1：

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

示例 2：

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

提示:

• 節點數在 [1, 104] 范圍內
• -105 <= Node.val <= 105

Python描述

遞歸計算sum，然後統計出現次數

# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, val=0, left=None, right=None):
# self.val = val
# self.left = left
# self.right = right
class Solution:
def findFrequentTreeSum(self, root: TreeNode) -> List[int]:
tmp = {
}
def get_sum(node):
if node is None:
return 0
s = get_sum(node.left) + get_sum(node.right) + node.val
tmp[s] = tmp.get(s,[]) + [node.val,]
return s
get_sum(root)
max_len = 0
for item in tmp.values():
max_len = max(max_len,len(item))
res = []
for key,item in tmp.items():
if len(item) == max_len:
res.append(key)
return res