1109.航班预订统计(Java题解)

1109.航班预订统计

暴力解法

笔者的最初想到的思路是对每次的预订记录进行累加，遍历每个二维数组的子元素求得答案。

class Solution {
    public int[] corpFlightBookings(int[][] bookings, int n) {
        int[] answer = new int[n];
        for(int i = 0;i < bookings.length;i++){
            int t = bookings[i][0]-1;
            while(t < bookings[i][1]){
                answer[t] = answer[t] + bookings[i][2];
                t++;
            }
        }
        return answer;
    }
}

暴力解法虽然思路清晰，但用时明显过长。(虽然通过）

差分数组

对于此题我们可以引入一个差分数组的概念。

我们拥有数组bookings，bookings[i]代表每份订单，每份订单由三部分组成：

• bookings[i][0]: 订单起始号。
• bookings[i][1]: 订单末尾号。
• bookings[i][2]: 订单人数。

我们假设新建一个长度为n的answer整型数组。answer的序号便为航班号。那么我们是不是可以对answer进行赋值，对应的航班开始处，给answer数组赋值。

answer[航班号-1] = answer[航班号-1] + bookings[i][1]

那么我们对其进行累和是不是能够得到相应位置的预定数呢？

answer[i] = answer[i] + answer[i-1]

这里我们除了在开始时给数组赋值，同样也要在结束时减去相应的值。

answer[结束时的航班号] = answer[结束时的航班号] - bookings[i][1]
//注意此处是结束时的航班号，因为对于answer数组来说，在结束时一份订单时，预订仍然存在。

完整代码：

class Solution {
    public int[] corpFlightBookings(int[][] bookings, int n) {
        int[] answer = new int[n];
        for(int i = 0;i < bookings.length;i++){
            answer[bookings[i][0] - 1] = answer[bookings[i][0] - 1] + bookings[i][2];
            if(bookings[i][1] < n)
            answer[bookings[i][1]] = answer[bookings[i][1]] - bookings[i][2];
        }
        for(int i = 1;i < n;i++){
            answer[i] = answer[i] + answer[i-1];
        }
        return answer;
    }
}

提升明显。

---

2021.8.31