LeetCode | 0113. 경로 합 II 합 경로 II [파이썬]

LeetCode 0113. 경로 합 II 총 중간 [II] [경로] [다시] 파이썬

문제

LeetCode

이진 트리와 합을 감안할 때, 각 경로의 합이 주어진 금액에 해당하는 모든 루트 - 투 - 잎 경로를 찾을 수 있습니다.

참고 : 리프는 자녀가없는 노드입니다.

예:

아래 이진 트리를 감안할 때 sum = 22,

      5
     / \
    4   8
   /   / \
  11  13  4
 /  \    / \
7    2  5   1

반환:

[
   [5,4,11,2],
   [5,8,4,5]
]

문제

전원 버튼

이진 트리와 대상 및 리프 노드의 루트 노드에서 모든 경로의 합을 찾을 수를 감안할 때 주어진 대상 경로와 동일합니다.

설명 : 잎 노드는 노드에 자식 노드가 없습니다.

예 :
다음 이진 트리를 감안할 때, 대상 및 금액 = (22)

          5
         / \
        4   8
       /   / \
      11  13  4
     /  \    / \
    7    2  5   1

반환 값 :

[
   [5,4,11,2],
   [5,8,4,5]
]

사고

역행

先序遍历二叉树，记录路径。
符合条件的加入 res 中。
回溯记得要删除当前节点。

시간 복잡도 : 이진 트리 노드의 O (N-), N-수.
공간 복잡도 : O (N-), 최악의 경우, 단일 연결리스트로 이진 트리 타락한.

Python3 코드

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution:
    def pathSum(self, root: TreeNode, sum: int) -> List[List[int]]:
        res, path = [], []

        # 先序遍历：根左右
        def recur(root, target):
            if not root:
                return
            
            path.append(root.val)
            target -= root.val
            # 找到路径
            if target == 0 and not root.left and not root.right:
                res.append(list(path))  # 复制了一个 path 加入到 res 中，这样修改 path 不影响 res
            recur(root.left, target)
            recur(root.right, target)
            # 向上回溯，需要删除当前节点
            path.pop()
        
        recur(root, sum)
        return res

GitHub의 링크

파이썬