思路
利用一个二维数组(数组行数为3,分别对应三种垃圾)记录垃圾数量,arr[0][i]表示第i个房子的金属、纸和或玻璃垃圾。
解题方法
将三种垃圾数量(值与时间相同)相加,最后对应垃圾车最远需要走到哪个房间,加上对应的花费时间数
Code
class Solution {
public int garbageCollection(String[] garbage, int[] travel) {
int n=garbage.length;
int arr[][]=new int[3][n];
int ans=0;
for(int i=0;i<n;i++){
int len=garbage[i].length();
int M=0,P=0,G=0;
for(int j=0;j<len;j++){
if(garbage[i].charAt(j)=='M')M++;
if(garbage[i].charAt(j)=='P')P++;
if(garbage[i].charAt(j)=='G')G++;
}
ans+=M+P+G;
arr[0][i]=M;
arr[1][i]=P;
arr[2][i]=G;
}
for(int i=n-1;i>0;i--){
if(arr[0][i]!=0){
for(int j=i-1;j>=0;j--){
ans+=travel[j];
}
break;
}
}
for(int i=n-1;i>0;i--){
if(arr[1][i]!=0){
for(int j=i-1;j>=0;j--){
ans+=travel[j];
}
break;
}
}
for(int i=n-1;i>0;i--){
if(arr[2][i]!=0){
for(int j=i-1;j>=0;j--){
ans+=travel[j];
}
break;
}
}
return ans;
}
}