LeetCode题解(面试04.12)：求和路径(Python)

长行

于 2020-11-15 07:19:49 发布

阅读量232

点赞数

分类专栏：刷题记录文章标签： leetcode python 树二叉树深度优先搜索

本文链接：https://blog.csdn.net/Changxing_J/article/details/109572012

版权

刷题记录专栏收录该内容

1966 篇文章 55 订阅

订阅专栏

本文介绍了一种使用深度优先搜索算法解决二叉树中路径和等于特定值的问题。算法的时间复杂度为O(N^2)，空间复杂度也为O(N^2)，并提供了Python实现及执行时间。重点在于理解DFS在查找满足条件路径上的应用。

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

题目：原题链接（中等）

标签：树、二叉树、深度优先搜索

解法	时间复杂度	空间复杂度	执行用时
Ans 1 (Python)	$O(N^2)$	$O (N)$	132ms (67.79%)
Ans 2 (Python)
Ans 3 (Python)

解法一（深度优先搜索）：

class Solution:
    def __init__(self):
        self.ans = 0
        self.sum = 0
        self.total = 0
        self.lst = []

    def pathSum(self, root: TreeNode, sum: int) -> int:
        if not root:
            return 0

        self.sum = sum

        self.dfs(root)

        return self.ans

    def dfs(self, node):
        self.total += node.val
        self.lst.append(node.val)

        last = 0
        for i in range(len(self.lst) - 1, -1, -1):
            last += self.lst[i]
            if last == self.sum:
                self.ans += 1

        if node.left:
            self.dfs(node.left)
        if node.right:
            self.dfs(node.right)

        # print(self.total, self.lst, self.ans)

        self.lst.pop()
        self.total -= node.val