力扣2391---收集垃圾的最少总时间（Java、前缀和）

目录

题目描述：

思路描述：

代码：

        最笨的方法：

        较好的方法：

---

题目描述：

给你一个下标从 0 开始的字符串数组 garbage ，其中 garbage[i] 表示第 i 个房子的垃圾集合。garbage[i] 只包含字符 'M' ，'P' 和 'G' ，但可能包含多个相同字符，每个字符分别表示一单位的金属、纸和玻璃。垃圾车收拾 一 单位的任何一种垃圾都需要花费 1 分钟。

同时给你一个下标从 0 开始的整数数组 travel ，其中 travel[i] 是垃圾车从房子 i 行驶到房子 i + 1 需要的分钟数。

城市里总共有三辆垃圾车，分别收拾三种垃圾。每辆垃圾车都从房子 0 出发，按顺序 到达每一栋房子。但它们 不是必须 到达所有的房子。

任何时刻只有 一辆 垃圾车处在使用状态。当一辆垃圾车在行驶或者收拾垃圾的时候，另外两辆车 不能 做任何事情。

请你返回收拾完所有垃圾需要花费的 最少 总分钟数。

示例 1：

输入：garbage = ["G","P","GP","GG"], travel = [2,4,3]
输出：21
解释：
收拾纸的垃圾车：
1. 从房子 0 行驶到房子 1
2. 收拾房子 1 的纸垃圾
3. 从房子 1 行驶到房子 2
4. 收拾房子 2 的纸垃圾
收拾纸的垃圾车总共花费 8 分钟收拾完所有的纸垃圾。
收拾玻璃的垃圾车：
1. 收拾房子 0 的玻璃垃圾
2. 从房子 0 行驶到房子 1
3. 从房子 1 行驶到房子 2
4. 收拾房子 2 的玻璃垃圾
5. 从房子 2 行驶到房子 3
6. 收拾房子 3 的玻璃垃圾
收拾玻璃的垃圾车总共花费 13 分钟收拾完所有的玻璃垃圾。
由于没有金属垃圾，收拾金属的垃圾车不需要花费任何时间。
所以总共花费 8 + 13 = 21 分钟收拾完所有垃圾。

示例 2：

输入：garbage = ["MMM","PGM","GP"], travel = [3,10]
输出：37
解释：
收拾金属的垃圾车花费 7 分钟收拾完所有的金属垃圾。
收拾纸的垃圾车花费 15 分钟收拾完所有的纸垃圾。
收拾玻璃的垃圾车花费 15 分钟收拾完所有的玻璃垃圾。
总共花费 7 + 15 + 15 = 37 分钟收拾完所有的垃圾。

思路描述：

        依据题目的意思，我们可以得出收集垃圾的最少总时间就是让每个车不浪费多余的时间，即某一个车到达一个位置后，如果该位置以后都没有属于自己收集的垃圾类型，就不再消耗时间去遍历。

        最笨的方法就是多次遍历得到结果。

        较好的方法就是一次遍历得到结果，在计算三辆车总的时间时，对于第一部分，可直接对 garbage\textit{garbage}garbage 中所有字符串的长度求和；对于第二部分，可以求出每辆车到达每类垃圾最后一次出现的位置所消耗的时间，最后进行求和。

        在第二部分进行代码实现时，可用一个变量维护 travel的前缀和。

代码：

        最笨的方法：

public int garbageCollection(String[] garbage, int[] travel) {
        int ans=0;
        int n=garbage.length;
        int m=travel.length;
        int[] lastIndex={-1,-1,-1};
        int[] threeSum={0,0,0};
        List<Map<Character,Integer>> list=new ArrayList<>();
        for(int i=0;i<n;i++){
            if (garbage[i].contains("M")) {
                lastIndex[0]=i;
            }
            if (garbage[i].contains("P")) {
                lastIndex[1]=i;
            }
            if (garbage[i].contains("G")) {
                lastIndex[2]=i;
            }
            char[] charArray = garbage[i].toCharArray();
            int len=charArray.length;
            Map<Character,Integer> map=new HashMap<>();
            map.put('M',0);
            map.put('P',0);
            map.put('G',0);
            for (int j=0;j<len;j++){
                map.put(charArray[j],map.get(charArray[j])+1);
            }
            list.add(map);
        }
        for (int i = 0; i < 3; i++) {
            if(lastIndex[i]==-1){
                continue;
            } else if (lastIndex[i] == 0) {
                Map<Character, Integer> map = list.get(0);
                if(i==0){
                    ans+=map.get('M');
                } else if (i==1) {
                    ans+=map.get('P');
                }else{
                    ans+=map.get('G');
                }
                continue;
            }
            for(int j=0;j<lastIndex[i];){
                Map<Character, Integer> map = list.get(j);
                if(i==0){
                    ans+=map.get('M');
                } else if (i==1) {
                    ans+=map.get('P');
                }else{
                    ans+=map.get('G');
                }
                ans+=travel[j];
                j++;
                if(j==lastIndex[i]){
                    Map<Character, Integer> map2 = list.get(j);
                    if(i==0){
                        ans+=map2.get('M');
                    } else if (i==1) {
                        ans+=map2.get('P');
                    }else{
                        ans+=map2.get('G');
                    }
                }
            }
        }
        return ans;
    }

        较好的方法：

class Solution {
    public int garbageCollection(String[] garbage, int[] travel) {
        int ans=0;
        int n=garbage.length;
        int[] distance=new int[n];
        int[] lastIndex={-1,-1,-1};
        for(int i=0;i<n;i++){
            if(i>0){
                distance[i]=distance[i-1]+travel[i-1];
            }
            ans+=garbage[i].length();
            if (garbage[i].contains("M")) {
                lastIndex[0]=i;
            }
            if (garbage[i].contains("P")) {
                lastIndex[1]=i;
            }
            if (garbage[i].contains("G")) {
                lastIndex[2]=i;
            }
        }
        for(int i=0;i<3;i++){
            if(lastIndex[i]==-1){
                continue;
            }else{
                ans+=distance[lastIndex[i]];
            }
        }
        return ans;
    }
}