Leetcode:257/113 二叉树的所有路径

给定一个二叉树,返回所有从根节点到叶子节点的路径。

说明: 叶子节点是指没有子节点的节点。

示例:

输入:

   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

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值