给你一棵有 n 个节点的无向树,节点编号为 0 到 n-1 ,它们中有一些节点有苹果。 无向树的边由 edges 给出,其中 edges[i] = [fromi, toi] ,表示有一条边连接 from 和 toi 。 示例 1: 示例 2: 来源:力扣(LeetCode) 360 ms 56.3 MB
1. 题目
通过树上的一条边,需要花费 1 秒钟。
你从 节点 0 出发,请你返回最少需要多少秒,可以收集到所有苹果,并回到节点 0 。
除此以外,还有一个布尔数组 hasApple ,其中 hasApple[i] = true 代表节点 i 有一个苹果,否则,节点 i 没有苹果。
输入:n = 7, edges = [[0,1],[0,2],[1,4],[1,5],[2,3],[2,6]], hasApple = [false,false,true,false,true,true,false] 输出:8 解释:上图展示了给定的树,其中红色节点表示有苹果。 一个能收集到所有苹果的最优方案由绿色箭头表示。
输入: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 提示: 1 <= n <= 10^5 edges.length == n-1 edges[i].length == 2 0 <= fromi, toi <= n-1 fromi < toi hasApple.length == n
链接:https://leetcode-cn.com/problems/minimum-time-to-collect-all-apples-in-a-tree
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。2. 解题
class Solution { public: int minTime(int n, vector<vector<int>>& edges, vector<bool>& hasApple) { unordered_map<int,int> up;//向上走的路径 for(vector<int> & e : edges) up[e[1]] = e[0]; int s = 0; for(int i = 0; i < hasApple.size(); ++i) { if(hasApple[i]) dfs(i, up, s); } return 2*s; } void dfs(int i, unordered_map<int,int>& up, int& s) { if(up.count(i)) { s++; int to = up[i];//上层节点 up.erase(i);//删除边 dfs(to,up,s); } } };
本网页所有视频内容由 imoviebox边看边下-网页视频下载, iurlBox网页地址收藏管理器 下载并得到。
ImovieBox网页视频下载器 下载地址: ImovieBox网页视频下载器-最新版本下载
本文章由: imapbox邮箱云存储,邮箱网盘,ImageBox 图片批量下载器,网页图片批量下载专家,网页图片批量下载器,获取到文章图片,imoviebox网页视频批量下载器,下载视频内容,为您提供.
阅读和此文章类似的: 全球云计算