Python程式：查詢二叉樹中最頻繁的子樹和

假設我們有一個二叉樹，我們需要找到出現頻率最高的子樹和。節點的子樹和是該節點下所有值的總和，包括節點本身。

因此，如果輸入如下所示：

則輸出將為 3，因為它出現了兩次——一次作為左葉子節點，一次作為 3 - 6 + 6 的和。

為了解決這個問題，我們將遵循以下步驟：

  • count := 一個空對映
  • 定義一個函式 getSum()。這將接收節點作為引數。
  • 如果節點為空，則
    • 返回 0
  • mySum := getSum(節點的左子節點) + getSum(節點的右子節點) + 節點的值
  • count[mySum] := count[mySum] + 1
  • 返回 mySum
  • 在主方法中執行以下操作：
  • getSum(根節點)

讓我們看看下面的實現來更好地理解：

示例

 線上演示

from collections import defaultdict
class TreeNode:
   def __init__(self, data, left = None, right = None):
      self.val = data
      self.left = left
      self.right = right
class Solution:
   def solve(self, root):
      count = defaultdict(int)
      def getSum(node):
         if not node:
            return 0
         mySum = getSum(node.left) + getSum(node.right) + node.val
         count[mySum] += 1
         return mySum
      getSum(root)
      return max(count, key=count.get)
ob = Solution()
root = TreeNode(-6)
root.left = TreeNode(3)
root.right = TreeNode(6) print(ob.solve(root))

輸入

root = TreeNode(-6)
root.left = TreeNode(3)
root.right = TreeNode(6)

輸出

3

更新於：2020年10月19日

147 次檢視

開啟您的職業生涯

完成課程獲得認證

開始學習
廣告
© . All rights reserved.