一、题目
1、题目描述
给你一个下标从 0 开始的字符串数组
garbage
,其中garbage[i]
表示第i
个房子的垃圾集合。garbage[i]
只包含字符'M'
,'P'
和'G'
,但可能包含多个相同字符,每个字符分别表示一单位的金属、纸和玻璃。垃圾车收拾 一 单位的任何一种垃圾都需要花费1
分钟。同时给你一个下标从 0 开始的整数数组
travel
,其中travel[i]
是垃圾车从房子i
行驶到房子i + 1
需要的分钟数。城市里总共有三辆垃圾车,分别收拾三种垃圾。每辆垃圾车都从房子
0
出发,按顺序 到达每一栋房子。但它们 不是必须 到达所有的房子。任何时刻只有 一辆 垃圾车处在使用状态。当一辆垃圾车在行驶或者收拾垃圾的时候,另外两辆车 不能 做任何事情。
请你返回收拾完所有垃圾需要花费的 最少 总分钟数。
2、接口描述
python3
python
class Solution:
def garbageCollection(self, garbage: List[str], travel: List[int]) -> int:
cpp
cpp
class Solution {
public:
int garbageCollection(vector<string>& garbage, vector<int>& travel) {
}
};
3、原题链接
二、解题报告
1、思路分析
花费来自于:捡取所有垃圾 和 每个车所需的移动距离
就是一个模拟题,遍历garbage,如果某个垃圾种类出现,对应的垃圾车的移动距离更新下
2、复杂度
时间复杂度: O(n)空间复杂度:O(1)
3、代码详解
python3
python
class Solution:
def garbageCollection(self, garbage: List[str], travel: List[int]) -> int:
c1, c2, c3 = 0, 0, 0
s = 0
pre = 0
travel = [0] + travel
for i, x in enumerate(garbage):
s += len(x)
pre += travel[i]
if 'M' in x:
c1 = pre
if 'P' in x:
c2 = pre
if 'G' in x:
c3 = pre
return c1 + c2 + c3 + s
cpp
cpp
class Solution {
public:
int garbageCollection(vector<string>& garbage, vector<int>& travel) {
int c1 = 0, c2 = 0, c3 = 0, s = 0;
for (int i = 0, pre = 0, n = garbage.size(); i < n; i ++) {
s += garbage[i].size();
pre += i ? travel[i - 1] : 0;
if(~garbage[i].find('M'))
c1 = pre;
if(~garbage[i].find('P'))
c2 = pre;
if(~garbage[i].find('G'))
c3 = pre;
}
return s + c1 + c2 + c3;
}
};