算法设计与分析—二叉树中和为某一值的路径

题目描述:

二叉树中和为某一值的路径
给你二叉树的根节点 root 和一个整数目标和 targetSum ,找出所有 从根节点到叶子节点 路径总和等于给定目标和的路径。
叶子节点 是指没有子节点的节点。

示例1:
输入:root = [5,4,8,11,null,13,4,7,2,null,null,5,1],
targetSum = 22
输出:[[5,4,11,2],[5,8,4,5]]
请添加图片描述

示例 2:
输入:root = [1,2,3], targetSum = 5
输出:[]
请添加图片描述

示例 3:
输入:root = [1,2], targetSum = 0
输出:[]

来源:力扣(LeetCode)


算法实现:

解题思路
回溯法
使用二叉树的先序遍历访问每一个结点,访问到的结点放入栈中,每次判断一下,栈中的值是否为target (判断方法可以用target==0代替)。
每次递归回退时,都要出栈一次也达到回退的效果。
请添加图片描述

def pathSum(self, root, target):
    """
    :type root: TreeNode
    :type target: int
    :rtype: List[List[int]]
    """
    result=[]
    nums=[]
    def dfs(root,target):
        if root==None:
            return
        nums.append(root.val)
        target-=root.val
        if root.left==None and root.right==None and target==0:
            result.append(nums[:])
        dfs(root.left,target)
        dfs(root.right,target)
        nums.pop()
    dfs(root,target)
    return result


时间复杂度:O(n^2) ,空间复杂度O(n)

Guess you like

Origin blog.csdn.net/qq_39740279/article/details/120833432