Python에서 트리의 모든 값이 동일한지 확인하는 프로그램

바이너리 트리가 있다고 가정하면 트리의 모든 노드가 동일한 값을 갖는지 확인해야 합니다.

따라서 입력이 다음과 같으면

그러면 출력은 True

이 문제를 해결하기 위해 다음 단계를 따릅니다. −

• solve() 함수를 정의합니다. 이것은 뿌리를 내리고 val

• 루트가 null이면

  • 참을 반환

• val이 정의되지 않은 경우

  • val :=루트 값

• root의 값이 val과 같을 때 true를 반환하고 solve(root의 왼쪽, val) 및 solve(root의 오른쪽, val)도 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