收集垃圾的最少总时间- （LeetCode）-CSDN博客

本文链接：https://blog.csdn.net/xuyongyi/article/details/138733846

题目

给你一个下标从 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 分钟收拾完所有的垃圾。

提示

2 <= garbage.length <= 105
garbage[i] 只包含字母 'M' ，'P' 和 'G' 。
1 <= garbage[i].length <= 10
travel.length == garbage.length - 1
1 <= travel[i] <= 100

`解题思路`

遍历garbage，记录'M''P''G'的出现次数和'M''P''G'的最后出现的位置即可。又因为每个字符串中只有上述三个字母，因此可直接加上字符串的全部长度。

注意最后一个数组不需要判断和处理。

解题代码

class Solution {
    public int garbageCollection(String[] garbage, int[] travel) {
        String[] carName = new String[]{"M", "P", "G"};
        int[] cardValue = new int[]{0, 0, 0};
        int len = garbage.length;
        for (int c = 0; c < carName.length; c++) {
            int l = 0;
            for (int i = 0; i < len; i++) {
                for (int j = 0; j < garbage[i].length(); j++) {
                    //单位的任何一种垃圾都需要花费1分钟
                    char s = garbage[i].charAt(j);
                    if (carName[c].equals(String.valueOf(s))) {
                        cardValue[c] += 1;
                    }
                }
                boolean isCheck = false;
                for (int k = i + 1; k < len; k++) {
                    if (garbage[k].contains(carName[c])) {
                        isCheck = true;
                        break;
                    }
                }
                //判断是不是最后一个数组
                if (i != garbage.length - 1) {

                    if (isCheck) {
                        //判断下一个,获取去的路途
                        int x = travel[l];//前往一个垃圾站需要的时间
                        cardValue[c] += x;
                        l++;
                        if (l >= travel.length) {
                            l = 0;
                        }
                    }
                }
            }
        }
        return Arrays.stream(cardValue).sum();
    }
}