给定一个二叉树,返回所有从根节点到叶子节点的路径。
说明: 叶子节点是指没有子节点的节点。
示例:
输入: 1 / \ 2 3 \ 5 输出: ["1->2->5", "1->3"] 解释: 所有根节点到叶子节点的路径为: 1->2->5, 1->3
tips:遍历二叉树的所有路径,最直接的便是深度优先搜索(DFS),与先序遍历不同的是,我们需要每个从根节点到叶子节点的完整路径。需要注意,右子树的路径一定不包含左子树的节点。
我们采用递归进行DFS,首先将根节点放入路径中,依次遍历其左右节点,若存在,递归;若都不存在,打印当前路径作为一条完整路径。每次迭代保存当前路径,以当前路径列表与下一个节点进行拼接。此处解释为何用拼接而不用append,使用append会改变当前迭代的后续路径,而使用拼接会保持在该次迭代的全过程当前路径不变。代码如下:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def binaryTreePaths(self, root: TreeNode) -> List[str]:
if not root:
return []
cur=[root.val]
res=[]
def DFS(root,cur):
if not root.left and not root.right:
res.append(cur[:])
return
if root.left:
DFS(root.left,cur+[root.left.val])#注意此处不能为DFS(root.left,cur.append(root.left.val))
if root.right:
DFS(root.right,cur+[root.right.val])#若为append,改变了右子树的当前路径
DFS(root,cur)
return ["->".join([str(j) for j in i]) for i in res]#str.join(list)要保证list元素为str()
113 给定一个二叉树和一个目标和,找到所有从根节点到叶子节点路径总和等于给定目标和的路径。
说明: 叶子节点是指没有子节点的节点。
示例:
给定如下二叉树,以及目标和sum = 22
,5 / \ 4 8 / / \ 11 13 4 / \ / \ 7 2 5 1返回:
[ [5,4,11,2], [5,8,4,5] ]该题在剑指offer中要求在返回值的list中,数组长度大的数组靠前。
tips:因为树的节点值并非限定于正整数,无法剪枝,故需要找出所以路径。代码如下:
# Definition for a binary tree node.
# class TreeNode:
# def __init__(self, x):
# self.val = x
# self.left = None
# self.right = None
class Solution:
def pathSum(self, root: TreeNode, sum: int) -> List[List[int]]:
if not root:
return []
res=[]
cur=[root.val]
def DFS(root,cur):
if not root.left and not root.right:
res.append(cur)
return
if root.left:
DFS(root.left,cur+[root.left.val])
if root.right:
DFS(root.right,cur+[root.right.val])
DFS(root,cur)
def suma(l):
s=0
for i in l:
s+=i
return s
ress=[]
for i in res:
if suma(i)==sum:
ress.append(i)
ress.sort(key=lambda x:len(x),reverse=1)
return ress