剑指offer24：二叉树中和为某一值的路径

最新推荐文章于 2021-10-05 21:32:38 发布

sa726663676

最新推荐文章于 2021-10-05 21:32:38 发布

阅读量133

点赞数

分类专栏：剑指offer

本文链接：https://blog.csdn.net/sa726663676/article/details/113902981

版权

剑指offer 专栏收录该内容

11 篇文章 0 订阅

订阅专栏

该博客介绍了如何在给定二叉树中，按字典序找到节点值之和等于特定整数的所有路径。采用两种递归方法实现，一种利用共享列表减少内存消耗，另一种为每层递归增加额外存储。思路是深度优先搜索，回溯找到符合条件的路径。

摘要由CSDN通过智能技术生成

题目描述

输入一颗二叉树的根节点和一个整数，按字典序打印出二叉树中结点值的和为输入整数的所有路径。路径定义为从树的根结点开始往下一直到叶结点所经过的结点形成一条路径。

示例1

输入

{10,5,12,4,7},22

返回值

[[10,5,7],[10,12]]

思路：使用带追溯路径的递归方法。需要用到两个变量，一个变量用于追溯递归路径，一个变量用于记录满足条件的所有结果。

1.通过一个共享列表的pop和append的实现。优点：所有递归的栈都使用同一个列表可以减少内存消耗。

# -*- 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
        res = []
        path = []
        def recur(root, tar):
            if not root:return
            path.append(root.val)
            tar -= root.val
            if (tar == 0) and not root.left and not root.right:
                res.append(list(path))
            recur(root.left, tar)
            recur(root.right, tar)
            path.pop()
        recur(root, expectNumber)
        return res

2.通过为每一层递归加一个额外存储单元实现。好像没啥优点，缺点问题更大，尤其是当递归层数多的时候占用内存更大。

class Solution:
    # 返回二维列表，内部每个列表表示找到的路径
    def FindPath(self, root, expectNumber):
        if not root:
            return []
        result = []
        path = [root.val]
        self.sums = expectNumber
        self.DFS(root,result,path)
        return result
    def DFS(self,root,result,path):
        if root.left == None and root.right == None and sum(path) == self.sums:
            result.append(path)
        if root.left != None:
            self.DFS(root.left,result,path+[root.left.val])
        if root.right != None:
            self.DFS(root.right,result,path+[root.right.val])

#题目来源于牛客网：https://www.nowcoder.com/practice/b736e784e3e34731af99065031301bca?tpId=13&rp=1&ru=%2Fta%2Fcoding-interviews&qru=%2Fta%2Fcoding-interviews%2Fquestion-ranking&tab=answerKey