二叉树中和为某一值的路径 python

最新推荐文章于 2021-11-26 13:23:15 发布

一个小迷糊66

最新推荐文章于 2021-11-26 13:23:15 发布

阅读量565

点赞数 2

CC 4.0 BY-SA版权

分类专栏：刷题

本文链接：https://blog.youkuaiyun.com/jianyuchen23/article/details/98038442

刷题专栏收录该内容

70 篇文章

订阅专栏

本文介绍了一种算法，用于查找二叉树中所有节点值之和等于给定整数的路径。通过递归遍历树结构，利用栈记录路径上的节点值，并在达到叶子节点且路径和等于目标值时保存路径。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

输入一颗二叉树的根节点和一个整数，打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。

注意说stack类似指针存储的，相当于全局的，要pop，cursum只是局部的，不用减去。。stack要深拷贝进res，否则res中的也会被pop掉。

class Solution:
	def FindPath(self, root, expectNumber):
		if not root:
			return []
		cursum = 0
		expect = expectNumber
		res = []
		stack = []
		self.Findpp(root, cursum, stack, expect, res)
		return res
	def Findpp(self, root, cursum, stack, expect, res):
		cursum += root.val
		stack.append(root.val)
		if(cursum == expect and root.left == None and root.right == None):
			a = copy.deepcopy(stack)
			res.append(a)  #或者res.append(stack[:])
		if root.left:
			self.Findpp(root.left, cursum, stack, expect, res)
	    if root.right:
	    	self.Findpp(root.right, cursum, stack, expect, res)
	    stack.pop()