파이썬에서 이진 트리의 경로로 구성된 모든 숫자의 합을 찾는 프로그램

<시간/>

각 노드가 0에서 9까지의 단일 숫자를 포함하는 이진 트리가 있다고 가정합니다. 이제 루트에서 잎까지의 각 경로는 숫자를 순서대로 숫자로 나타냅니다. 트리의 모든 경로가 나타내는 숫자의 합을 찾아야 합니다.

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

출력은 46(4 → 6), 432(4 → 3 → 2), 435(4 → 3 → 5)로 680이 되며 합계는 913입니다.

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

solve() 함수를 정의합니다. 이것은 루트를 취합니다. string:=공백 문자열
root.left가 아닌 root.right가 0이 아닌 경우
- 반환 int(문자열 + str(루트 값))
총계:=0
루트의 왼쪽이 null이 아니면
- total :=total + solve의 숫자 값(루트의 왼쪽, 루트의 문자열 연결 값)
루트의 권한이 null이 아니면
- total :=total + solve의 숫자 값(루트의 오른쪽, 루트의 문자열 연결 값)
총 수익

이해를 돕기 위해 다음 구현을 살펴보겠습니다. −

예시

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, string=""):
      if root and not root.left and not root.right:
         return int(string + str(root.val))

      total = 0
      if root.left:
         total += int(self.solve(root.left, string + str(root.val)))

      if root.right:
         total += int(self.solve(root.right, string + str(root.val)))

      return total

ob = Solution()
root = TreeNode(4)
root.left = TreeNode(6)
root.right = TreeNode(3)
root.right.left = TreeNode(2)
root.right.right = TreeNode(5)
print(ob.solve(root))

입력

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