LeetCode | 0113 ruta Ruta suma Suma II II [Python]

LeetCode 0113 Ruta Sum II Medio Total [II] [ruta] [volver] Python

Problema

LeetCode

Dado un árbol binario y una suma, encontrar todos los caminos de la raíz a la hoja, donde la suma de cada ruta es igual a la suma determinada.

Nota: Una hoja es un nodo que no tienen hijos.

Ejemplo:

Dado el árbol binario y a continuación sum = 22,

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

Regreso:

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

problema

botón de encendido

Dado un árbol binario y un objetivo y para hallar la suma de todos los caminos desde el nodo raíz del nodo hoja es igual a un objetivo y camino dado.

Descripción: nodo hoja es un nodo no tiene nodos secundarios.

Ejemplo:
Dado el siguiente árbol binario, y el objetivo y la suma = 22,

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

Devuelve:

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

pensamiento

backtracking

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

complejidad Tiempo: O (n-), n-número de nodos de árbol binario.
la complejidad de las instalaciones: O (n), el peor de los casos, un degenerado árbol binario en una sola lista enlazada.

Código 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

enlace GitHub

Pitón