LeetCode 1443. 收集树上所有苹果的最少时间（自底向上DFS）

Michael阿明

发布于 2020-07-13 06:23:34

50400

代码可运行

文章被收录于专栏：Michael阿明学习之路Michael阿明学习之路

运行总次数：0

代码可运行

1. 题目

给你一棵有 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
 
提示：
1 <= n <= 10^5
edges.length == n-1
edges[i].length == 2
0 <= fromi, toi <= n-1
fromi < toi
hasApple.length == n

来源：力扣（LeetCode）链接：https://leetcode-cn.com/problems/minimum-time-to-collect-all-apples-in-a-tree 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2. 解题

由题目条件可知向上走的路径只有1个分支，把反向的路径存在哈希map里
遍历hasApple数组，对有苹果的序号，进行dfs往上找，找到一条边，就在哈希表里删除一条
最后返回边的个数乘以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);
    	}
    }
};