题目描述
输入一颗二叉树的根节点和一个整数,打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。(注意: 在返回值的list中,数组长度大的数组靠前)
利用DFS思想
# -*- coding:utf-8 -*-
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
# 返回二维列表,内部每个列表表示找到的路径
def __init__(self):
self.result=[]
def FindPath(self, root, expectNumber):
# write code here
if not root:
return []
path=[]
self.dfs(root,expectNumber,path,0)
self.result.sort(key = lambda i:len(i),reverse=True)
return self.result
def dfs(self,root,expectNumber,path,count):
if not root:
return None
count+=root.val
if root.left==None and root.right==None:
if count==expectNumber:
self.result.append(path+[root.val])
else:
self.dfs(root.left,expectNumber,path+[root.val],count)
self.dfs(root.right,expectNumber,path+[root.val],count)
今日查漏补缺:
list的append方法直接在原list上修改,传参后使用append会导致结果错误,用+会返回新对象,不改变原值
参考:Python中append和+以及extend的区别