leetcode热题100.路径总和 III

简介: leetcode热题100.路径总和 III

题目

给定一个二叉树的根节点 root ,和一个整数 targetSum ,求该二叉树里节点值之和等于 targetSum 的 路径 的数目。

路径 不需要从根节点开始,也不需要在叶子节点结束,但是路径方向必须是向下的(只能从父节点到子节点)。

示例 1:

输入:root = [10,5,-3,3,2,null,11,3,-2,null,1], targetSum = 8

输出:3

解释:和等于 8 的路径有 3 条,如图所示。

示例 2:

输入:root = [5,4,8,11,null,13,4,7,2,null,null,5,1], targetSum = 22

输出:3

提示:

二叉树的节点个数的范围是 [0,1000]

− 1 0 9 < = N o d e . v a l < = 1 0 9 -10^9 <= Node.val <= 10^9109<=Node.val<=109

− 1000 < = t a r g e t S u m < = 1000 -1000 <= targetSum <= 10001000<=targetSum<=1000

思路1

先序遍历所有节点,对于每次遍历节点,从这个节点出发往下查询有无路径和为target,如果发现是target则答案加一,并且返回。

复杂度1

时间复杂度:

遍历了两个树节点 O ( n 2 ) O(n^2)O(n2)

空间复杂度:

只在栈上开辟了空间: O ( n ) O(n)O(n)

Code1

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def pathSum(self, root: Optional[TreeNode], targetSum: int) -> int:
        self.ans = 0
        def dfs(node,sum1):
            if sum1==targetSum:
                self.ans+=1 
            if node.left: dfs(node.left,sum1+node.left.val)
            if node.right: dfs(node.right,sum1+node.right.val)
        def findnode(root):
            if not root:return
            dfs(root,root.val)
            findnode(root.left)
            findnode(root.right)
        findnode(root)
        return self.ans

思路2

使用前缀树计算每个节点到根节点的差值,prefix是记录前缀和的字典,初始prefix【0】设置为1,因为对于路径长度为1的路径而言,如果正好等于target,也算作答案。

在dfs函数中,root为当前节点,cur是当前这条路径的下所有节点的和。我们注意到,当我们得出这条路径的和cur时,如果prefix【cur-targetSum】存在,说明这条路径上必然有若干点可以实现和为target的。

我们将其加入答案当中。

接下来,我们将cur加入前缀和中,开始递归子节点,在递归结束后要将prefix[cur] -= 1,这是由于我们只能统计往下的路径,但是树的遍历会同时搜索两个方向的子树。因此我们应当在搜索完以某个节点为根的左右子树之后,应当回溯地将路径总和从哈希表中删除,防止统计到跨越两个方向的路径。

复杂度2

时间复杂度:

遍历了两个树节点 O ( n 2 ) O(n^2)O(n2)

空间复杂度:

只在栈上开辟了空间: O ( n ) O(n)O(n)

Code2

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def pathSum(self, root: Optional[TreeNode], targetSum: int) -> int:
        prefix = defaultdict(int)
        prefix[0] = 1
        def dfs(root,cur):
            if not root: return 0
            ret = 0
            
            cur += root.val
            ret += prefix[cur-targetSum]
            prefix[cur] += 1
            ret += dfs(root.left,cur)
            ret += dfs(root.right,cur)
            prefix[cur] -= 1
        
            return ret
        
        return dfs(root,0)


目录
相关文章
|
1月前
【LeetCode 35】112.路径总和
【LeetCode 35】112.路径总和
23 0
|
4月前
|
算法 Unix 测试技术
力扣经典150题第五十二题:简化路径
力扣经典150题第五十二题:简化路径
42 0
|
1月前
【LeetCode 36】113.路径总和II
【LeetCode 36】113.路径总和II
28 0
|
3月前
|
机器人 Python
【Leetcode刷题Python】62. 不同路径
LeetCode 62题 "不同路径" 的Python解决方案,使用动态规划算法计算机器人从网格左上角到右下角的所有可能路径数量。
68 0
|
5月前
|
存储 SQL 算法
LeetCode题目113:多种算法实现 路径总和ll
LeetCode题目113:多种算法实现 路径总和ll
|
1月前
【LeetCode 34】257.二叉树的所有路径
【LeetCode 34】257.二叉树的所有路径
12 0
|
3月前
|
Python
【Leetcode刷题Python】113. 路径总和 II
LeetCode上113号问题"路径总和 II"的Python实现,通过深度优先搜索来找出所有从根节点到叶子节点路径总和等于给定目标和的路径。
41 3
【Leetcode刷题Python】113. 路径总和 II
|
3月前
|
存储 Python
【Leetcode刷题Python】64. 最小路径和
一种使用动态规划解决LeetCode上64题“最小路径和”的Python实现方法,通过维护一个一维数组来计算从网格左上角到右下角的最小路径总和。
32 1
【Leetcode刷题Python】64. 最小路径和
|
3月前
|
存储 算法 Linux
LeetCode第71题简化路径
文章讲述了LeetCode第71题"简化路径"的解题方法,利用栈的数据结构特性来处理路径中的"."和"..",实现路径的简化。
LeetCode第71题简化路径
|
3月前
|
算法
LeetCode第64题最小路径和
LeetCode第64题"最小路径和"的解题方法,运用动态规划思想,通过构建一个dp数组来记录到达每个点的最小路径和,从而高效求解。
LeetCode第64题最小路径和