Pythonでツリーのすべての要素の合計を見つけるプログラム

いくつかの値を含む二分木があるとすると、ツリー内のすべての値の合計を見つける必要があります。

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

その場合、出力は14になります

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

• 関数recurse（）を定義します。これはノードを取ります

• val：=ノードの値

• ノードの左側がnullでない場合、

  • val：=val + recurse（ノードの左側）

• ノードの権利がnullでない場合、

  • val：=val + recurse（ノードの右側）

• 戻り値

• メインの方法から、次のようにします-

• ルートがゼロ以外の場合、

  • 0を返す

• recurse（root）を返す

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

例

class TreeNode:
   def __init__(self, data, left = None, right = None):
      self.val = data
      self.left = left
      self.right = right
class Solution:
   def recurse(self, node):
      val = node.val
      if node.left:
         val += self.recurse(node.left)
      if node.right:
         val += self.recurse(node.right)
      return val
   def solve(self, root):
      if not root:
         return 0
      return self.recurse(root)
ob = Solution()
root = TreeNode(2)
root.right = TreeNode(4)
root.right.left = TreeNode(3)
root.right.right = TreeNode(5)
print(ob.solve(root))

入力

root = TreeNode(2)
root.right = TreeNode(4)
root.right.left = TreeNode(3)
root.right.right = TreeNode(5)

出力

14