列表

详情

1443. 收集树上所有苹果的最少时间

给你一棵有 n 个节点的无向树，节点编号为 0 到 n-1 ，它们中有一些节点有苹果。通过树上的一条边，需要花费 1 秒钟。你从 节点 0 出发，请你返回最少需要多少秒，可以收集到所有苹果，并回到节点 0 。

无向树的边由 edges 给出，其中 edges[i] = [fromi, toi] ，表示有一条边连接 from 和 toi 。除此以外，还有一个布尔数组 hasApple ，其中 hasApple[i] = true 代表节点 i 有一个苹果，否则，节点 i 没有苹果。

 

示例 1：

输入：n = 7, edges = [[0,1],[0,2],[1,4],[1,5],[2,3],[2,6]], hasApple = [false,false,true,false,true,true,false]
输出：8 
解释：上图展示了给定的树，其中红色节点表示有苹果。一个能收集到所有苹果的最优方案由绿色箭头表示。

示例 2：

输入：n = 7, edges = [[0,1],[0,2],[1,4],[1,5],[2,3],[2,6]], hasApple = [false,false,true,false,false,true,false]
输出：6
解释：上图展示了给定的树，其中红色节点表示有苹果。一个能收集到所有苹果的最优方案由绿色箭头表示。

示例 3：

输入：n = 7, edges = [[0,1],[0,2],[1,4],[1,5],[2,3],[2,6]], hasApple = [false,false,false,false,false,false,false]
输出：0

 

提示：

原站题解

去查看

上次编辑到这里，代码来自缓存 点击恢复默认模板
class Solution { public: int minTime(int n, vector<vector<int>>& edges, vector<bool>& hasApple) { } };

python3 解法, 执行用时: 224 ms, 内存消耗: 73.3 MB, 提交时间: 2023-08-11 11:10:36 

'''
关键点：将每个苹果对应的父节点标记为True
如图中，2/4/5为需要收集的苹果，途中必须经过0和1节点，
可以将需要收集的苹果的祖先节点的hasApple状态自下而上标记为True，
即hasApple[0]=hasApple[1]=True
遍历无向树edges中各个边，如果构成边的两点的hasApple状态都为True，
说明需要经过这条边，res+=1，而收集的过程中每条边需要走两次，所以2*res即为所求
'''
class Solution:
    def minTime(self, n: int, edges: List[List[int]], hasApple: List[bool]) -> int:
        graph = collections.defaultdict(set)
        for u,v in edges:
            graph[u].add(v)
            graph[v].add(u)
        
        visited=set()
        def dfs(root):
            visited.add(root)
            for nex in graph[root]:
                if nex not in visited:                    
                    dfs(nex)
                    if hasApple[nex]:
                        hasApple[root]=True
        dfs(0)
        res=0
        for u,v in edges:
            if hasApple[u] and hasApple[v]:
                res+=1
        return res*2

上一题

© 2024 leetcode.cn