python二叉树深度遍历_【python-lintcode376-树的深度遍历】二叉树的路径和

给定一个二叉树，找出所有路径中各节点相加总和等于给定 目标值 的路径。

一个有效的路径，指的是从根节点到叶节点的路径。

样例

样例1:

输入:

{1,2,4,2,3}

5

输出: [[1, 2, 2],[1, 4]]

说明:

这棵树如下图所示：

1

/ \

2 4

/ \

2 3

对于目标总和为5，很显然1 + 2 + 2 = 1 + 4 = 5

样例2:

输入:

{1,2,4,2,3}

3

输出: []

说明:

这棵树如下图所示：

1

/ \

2 4

/ \

2 3

注意到题目要求我们寻找从根节点到叶子节点的路径。

1 + 2 + 2 = 5, 1 + 2 + 3 = 6, 1 + 4 = 5

这里没有合法的路径满足和等于3.

"""

Definition of TreeNode:

class TreeNode:

def __init__(self, val):

self.val = val

self.left, self.right = None, None

"""

class Solution:

"""

@param: root: the root of binary tree

@param: target: An integer

@return: all valid paths

"""

def binaryTreePathSum(self, root, target):

# write your code here

global res

#保存结果

res=[]

#临时数组

tmp=[]

self.pathSum(res,tmp,root,target)

return res

def pathSum(self,res,tmp,root,target):

if root == None:

return

#首先将当前节点的值加入到tmp中

tmp.append(root.val)

#如果当前节点没有左右孩子，说明是叶子节点

if root.left == None and root.right == None:

#计算根节点到叶子节点的值

s = sum(tmp)

#如果该值等于target

if s == target:

#这里需要拷贝一份tmp，再加入到结果中

newtmp=tmp[:]

res.append(newtmp)

#如果有左孩子

if root.left!=None:

self.pathSum(res,tmp,root.left,target)

#如果遍历到叶子节点了，且不符合sum(tmp)==target，则最后面的节点出栈

tmp.pop()

#如果有右孩子

if root.right!=None:

self.pathSum(res,tmp,root.right,target)

#如果遍历到叶子节点了，且不符合sum(tmp)==target，则最后面的节点出栈

tmp.pop()

标签：tmp,lintcode376,target,python,res,self,二叉树,root,节点

来源： https://www.cnblogs.com/xiximayou/p/12396986.html