二叉树中和为某一值的路径
输入一颗二叉树的跟节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
思路
- 递归
递归的思想与数学归纳法不谋而合。在用递归的思想解题时,base_condition对应n=1的情形,这里指节点为叶节点,如果叶节点的值与期望的值相同,返回;递归条件的写法,类似于假设n=k-1成立,即已知根节点的左子树或右子树的总和等于expectNumber-root.val,那么n=k时的路径只需要加上root节点即可。
代码
# -*- coding:utf-8 -*-
class TreeNode:
def __init__(self, x):
self.val = x
self.left = None
self.right = None
class Solution:
# 返回二维列表,内部每个列表表示找到的路径
def FindPath(self, root, expectNumber):
# write code here
if not root:
return []
if root and root.left is None and root.right is None and expectNumber == root.val:
return [[root.val]]
res = []
left = self.FindPath(root.left,expectNumber-root.val)
right = self.FindPath(root.right,expectNumber-root.val)
for i in left+right:
res.append([root.val]+i)
return res