收集垃圾的最少总时间

/*
 * 思路：
 * 1. 使用流遍历每个房子的垃圾，计算收拾每种垃圾所需要的时间。
 * 2. 使用流计算每种垃圾车到达的最后一个房子位置，计算行驶时间。
 */
import java.util.stream.IntStream;

public class Solution {
    public int minTotalTime(String[] garbage, int[] travel) {
        int[] lastHouse = new int[3];
        int[] collectTime = new int[3];

        IntStream.range(0, garbage.length).forEach(i -> {
            garbage[i].chars().forEach(c -> {
                if (c == 'M') {
                    collectTime[0]++;
                    lastHouse[0] = i;
                } else if (c == 'P') {
                    collectTime[1]++;
                    lastHouse[1] = i;
                } else if (c == 'G') {
                    collectTime[2]++;
                    lastHouse[2] = i;
                }
            });
        });

        IntStream.range(1, garbage.length).forEach(i -> {
            if (i <= lastHouse[0]) {
                collectTime[0] += travel[i - 1];
            }
            if (i <= lastHouse[1]) {
                collectTime[1] += travel[i - 1];
            }
            if (i <= lastHouse[2]) {
                collectTime[2] += travel[i - 1];
            }
        });

        return IntStream.of(collectTime).sum();
    }
}