给你一个下标从 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
问题简要描述:返回收拾完所有垃圾需要花费的最少总分钟数
细节阐述:
- 哈希表 last 记录每种垃圾最后出现的房子下标
Java
class Solution {
public int garbageCollection(String[] garbage, int[] travel) {
Map<Character, Integer> last = new HashMap<>();
int ans = 0;
for (int i = 0; i < garbage.length; i++) {
ans += garbage[i].length();
for (char c : garbage[i].toCharArray()) {
last.put(c, i);
}
}
int ts = 0;
for (int i = 0; i < travel.length; i++) {
ts += travel[i];
for (int v : last.values()) {
if (v == i + 1) {
ans += ts;
}
}
}
return ans;
}
}
Python3
class Solution:
def garbageCollection(self, garbage: List[str], travel: List[int]) -> int:
last = {}
ans = 0
for i, g in enumerate(garbage):
ans += len(g)
for j, v in enumerate(g):
last[v] = i
ts = 0
for i, v in enumerate(travel):
ts += v
ans += sum(ts for j in last.values() if i + 1 == j)
return ans
TypeScript
function garbageCollection(garbage: string[], travel: number[]): number {
let last: Map<string, number> = new Map();
let ans = 0;
for (let i = 0; i < garbage.length; i++) {
ans += garbage[i].length;
for (const c of garbage[i]) {
last.set(c, i);
}
}
let ts = 0;
for (let i = 0; i < travel.length; i++) {
ts += travel[i];
for (const [_, v] of last) {
if (i + 1 == v) {
ans += ts;
}
}
}
return ans;
};
C++
class Solution {
public:
int garbageCollection(vector<string>& garbage, vector<int>& travel) {
unordered_map<int, int> last;
int ans = 0;
for (int i = 0; i < garbage.size(); i++){
ans += garbage[i].size();
for (char& c : garbage[i]) {
last[c] = i;
}
}
int ts = 0;
for (int i = 0; i < travel.size(); i++){
ts += travel[i];
for (auto& [k, v] : last) {
if (i + 1 == v) {
ans += ts;
}
}
}
return ans;
}
};