本文主要是介绍老卫带你学---leetcode刷题(437. 路径总和 III),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
437. 路径总和 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]
-109 <= Node.val <= 109
-1000 <= targetSum <= 1000
解决:
其实只需要在每次遍历中,保留路径的sum,并且count计数就可
class Solution:def pathSum(self, root: Optional[TreeNode], targetSum: int) -> int:def dfs(root,sum_list):if root is None: return 0sum_list = [num+root.val for num in sum_list] + [root.val]return sum_list.count(targetSum)+dfs(root.left,sum_list)+dfs(root.right,sum_list)return dfs(root,[])
这篇关于老卫带你学---leetcode刷题(437. 路径总和 III)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!