航班预订统计

/*
 * 题目思路：
 * 使用Java Stream的方式实现类似的逻辑。
 * 1. 使用一个初始大小为 n 的数组来存储结果。
 * 2. 使用流处理每一个预订记录。
 * 3. 在第一个索引处加 seats，在最后一个索引的后一个位置减 seats。
 * 4. 最后使用累加流来更新结果。
 */

import java.util.Arrays;

public int[] corpFlightBookings(int[][] bookings, int n) {
    int[] answer = new int[n];
    Arrays.stream(bookings).forEach(booking -> {
        int first = booking[0];
        int last = booking[1];
        int seats = booking[2];
        answer[first - 1] += seats;
        if (last < n) {
            answer[last] -= seats;
        }
    });
    // 累积求和
    Arrays.parallelPrefix(answer, Integer::sum);
    return answer;
}