Pythonでバイナリツリーの最も頻繁なサブツリーの合計を見つけるプログラム

二分木があるとすると、最も頻繁なサブツリーの合計を見つける必要があります。ノードのサブツリーの合計は、実際には、ノード自体を含む、ノードの下のすべての値の合計です。

したがって、入力が次のような場合

その場合、出力は2回発生するため3になります。1回は左の葉として、もう1回は3-6+6の合計として発生します。

これを解決するには、次の手順に従います-

• count：=空のマップ
• 関数getSum（）を定義します。これはノードを取ります
• ノードがnullの場合、
  • 0を返す
• mySum：=getSum（ノードの左側）+ getSum（ノードの右側）+ノードの値
• count [mySum]：=count [mySum] + 1
• mySumを返す
• メインの方法から次の手順を実行します
• getSum（root）

理解を深めるために、次の実装を見てみましょう-

例

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