LeetCode题解(1273)：删除树节点(Python)

最新推荐文章于 2022-06-22 16:52:03 发布

长行

最新推荐文章于 2022-06-22 16:52:03 发布

阅读量417

点赞数

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

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

版权

刷题记录专栏收录该内容

1966 篇文章 55 订阅

订阅专栏

该博客探讨了一种使用深度优先搜索(DFS)解决关于删除树节点的问题。算法以时间复杂度O(N)和空间复杂度O(N)运行，并提供了三种不同的Python实现。解法一中，通过构建图并从根节点开始递归地进行DFS，计算剩余节点的数量。

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

题目：原题链接（中等）

标签：深度优先搜索

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

解法一：

class Solution:
    def deleteTreeNodes(self, nodes: int, parent: List[int], value: List[int]) -> int:
        graph = collections.defaultdict(set)

        for i in range(nodes):
            graph[parent[i]].add(i)

        root = graph[-1].pop()

        def dfs(node):
            """返回 剩余节点数,和"""
            num0 = 1
            val0 = value[node]

            for nxt in graph[node]:
                num1, val1 = dfs(nxt)
                num0 += num1
                val0 += val1

            if val0 == 0:
                return 0, 0
            else:
                return num0, val0

        return dfs(root)[0]