守护太空城

// 题目思路：
// 使用Java Stream可以简化对数组的处理。
// 我们仍然需要追踪每个舱室的屏障状态和能量消耗。

import java.util.Arrays;

public class MeteorShowerProtectionStream {

    public int minEnergy(int[] time, int[] position) {
        int n = time.length;
        int maxPosition = 101; // 最大舱室数
        int[] energy = new int[maxPosition];

        // 用Stream来简化对数组的遍历
        Arrays.stream(position).distinct().forEach(p -> {
            int maxTime = Arrays.stream(time).filter(t -> position[Arrays.binarySearch(position, p)] == p).max().orElse(0);
            energy[p] = maxTime;
        });

        int totalEnergy = Arrays.stream(energy).sum();
        return totalEnergy;
    }

    public static void main(String[] args) {
        MeteorShowerProtectionStream msp = new MeteorShowerProtectionStream();
        int[] time = {1, 2, 1};
        int[] position = {6, 3, 3};
        System.out.println(msp.minEnergy(time, position)); // 输出：5
    }
}