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

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

五月 23, 2020

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

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

无向树的边由 edges 给出，其中 edges[i] = [fromi, toi] ，表示有一条边连接 from 和 toi 。除此以外，还有一个布尔数组 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
解释：上图展示了给定的树，其中红色节点表示有苹果。一个能收集到所有苹果的最优方案由绿色箭头表示。

解法

当时做题时候没有想到更好的办法处理那个二维数组，每次都遍历，最后就会超时，可惜了。
看到讨论区用map和list存了一下对应的父节点和子节点，每次都能直接取出对应的子节点，按照我最开始的思路改的。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
class Solution {
    private Map<Integer, List<Integer>> map;
    public int minTime(int n, int[][] edges, List<Boolean> hasApple) {
        map = new HashMap<>(n);
        for (int[] e : edges) {
            List<Integer> ch = map.get(e[0]);
            if (ch == null)
                map.put(e[0], ch = new ArrayList<>());
            ch.add(e[1]);
        }
        fun(0,0,hasApple);
        return ans;
    }

    public int ans = 0;
    public int fun(int root,int depth, List<Boolean> hasApple) {//每次返回深度
        List<Integer> list = map.get(root);
        if (list == null) {
           //是根节点
            return hasApple.get(root) ? depth : -1;
        }
        int flag = -1;//flag表示这棵树的子树有没有苹果
        for (Integer i : list) {
            int re = fun(i,depth + 1,hasApple);    
            if (re != -1) {
                flag = 1;
                ans += ((re - depth)*2);
            }
        }
        if (flag == -1) {//子树没有苹果还得看当前根节点有没有苹果
            return hasApple.get(root) ? depth : -1;
        } else {
            return depth;
        }
        
    }
}

leetcode 98/100