1443. Minimum Time to Collect All Apples in a Tree

Posted Veritas des Liberty

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了1443. Minimum Time to Collect All Apples in a Tree相关的知识,希望对你有一定的参考价值。

Given an undirected tree consisting of n vertices numbered from 0 to n-1, which has some apples in their vertices. You spend 1 second to walk over one edge of the tree. Return the minimum time in seconds you have to spend in order to collect all apples in the tree starting at vertex 0 and coming back to this vertex.

The edges of the undirected tree are given in the array edges, where edges[i] = [fromi, toi] means that exists an edge connecting the vertices fromi and toi. Additionally, there is a boolean array hasApple, where hasApple[i] = true means that vertex i has an apple, otherwise, it does not have any apple.

 

Example 1:

技术图片

Input: n = 7, edges = [[0,1],[0,2],[1,4],[1,5],[2,3],[2,6]], hasApple = [false,false,true,false,true,true,false]
Output: 8 
Explanation: The figure above represents the given tree where red vertices have an apple. One optimal path to collect all apples is shown by the green arrows.  

Example 2:

技术图片

Input: n = 7, edges = [[0,1],[0,2],[1,4],[1,5],[2,3],[2,6]], hasApple = [false,false,true,false,false,true,false]
Output: 6
Explanation: The figure above represents the given tree where red vertices have an apple. One optimal path to collect all apples is shown by the green arrows.  

Example 3:

Input: n = 7, edges = [[0,1],[0,2],[1,4],[1,5],[2,3],[2,6]], hasApple = [false,false,false,false,false,false,false]
Output: 0

 

Constraints:

  • 1 <= n <= 10^5
  • edges.length == n-1
  • edges[i].length == 2
  • 0 <= fromi, toi <= n-1
  • fromi < toi
  • hasApple.length == n

 

题意:

  给出一棵树,树中某些节点有苹果,求从根节点出发,将所有的苹果收集完,并返回根节点所需的步数。

思路:

  这是一道DFS的题目,可以把树看成是一个图,然后用DFS遍历图,记录下遍历所需的步数,因为需要返回所以在寻找苹果的过程中,应该将所需的步数 * 2 。这里我们用递归的方法来遍历树,如果子树中没有苹果,则这条遍历路径的步数应该置为0. 否则将遍历到该点所需的步数累加到总步数中。

Code:

 1 class Solution {
 2 public:
 3     vector<vector<int> > grap;
 4 
 5     int DFS(int index, int mySteps, vector<bool>& hasApple) {
 6         int childrenSteps = 0;
 7         for (int i : grap[index]) {
 8             childrenSteps += DFS(i, 2, hasApple);
 9         }
10         if (childrenSteps == 0 && hasApple[index] == false)
11             return 0;
12         return childrenSteps + mySteps;
13         
14     }
15     int minTime(int n, vector<vector<int>>& edges, vector<bool>& hasApple) {
16         grap.resize(n + 1);
17         for (int i = 0; i < edges.size(); ++i) 
18             grap[edges[i][0]].push_back(edges[i][1]);
19         return DFS(0, 0, hasApple);
20     }
21 };

 

参考:

  https://leetcode.com/problems/minimum-time-to-collect-all-apples-in-a-tree/discuss/623673/C%2B%2B-Java-Detailed-explanation-with-a-Picture-for-visualization

以上是关于1443. Minimum Time to Collect All Apples in a Tree的主要内容,如果未能解决你的问题,请参考以下文章

LeetCode 2187. Minimum Time to Complete Trips

lc 1199. Minimum Time to Build Blocks

algorithm@ O(3/2 n) time to findmaximum and minimum in a array

LeetCode 1723. 完成所有工作的最短时间 Find Minimum Time to Finish All Jobs(Java)

539. Minimum Time Difference 最小时差

539. Minimum Time Difference