ツリー内のすべての値がPythonで同じかどうかをチェックするプログラム

二分木があるとすると、ツリー内のすべてのノードが同じ値であるかどうかを確認する必要があります。

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

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

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

• 関数solve（）を定義します。これはルートになり、val

• ルートがnullの場合、

  • Trueを返す

• valが定義されていない場合、

  • val：=ルートの値

• ルートの値がvalと同じで、solve（ルートの左側、val）およびsolve（ルートの右側、val）もtrueの場合、trueを返します

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

例

class TreeNode:
   def __init__(self, val, left=None, right=None):
      self.val = val
      self.left = left
      self.right = right
class Solution:
   def solve(self, root, val=None):
      if not root:
         return True
      if val is None:
         val = root.val
      return root.val == val and self.solve(root.left, val) and
self.solve(root.right, val)
ob = Solution()
root = TreeNode(5)
root.left = TreeNode(5)
root.right = TreeNode(5)
root.left.left = TreeNode(5)
root.left.right = TreeNode(5)
print(ob.solve(root))

入力

root = TreeNode(5)
root.left = TreeNode(5)
root.right = TreeNode(5)
root.left.left = TreeNode(5)
root.left.right = TreeNode(5)

出力

True