1109. 航班预订统计 差分思想的运用

链接：https://leetcode-cn.com/problems/corporate-flight-bookings

这里有 n 个航班，它们分别从 1 到 n 进行编号。

有一份航班预订表 bookings ，表中第 i 条预订记录 bookings[i] = [firsti, lasti, seatsi] 意味着在从 firsti 到 lasti （包含 firsti 和 lasti ）的 每个航班 上预订了 seatsi 个座位。

请你返回一个长度为 n 的数组 answer，里面的元素是每个航班预定的座位总数。

示例 1：

输入：bookings = [[1,2,10],[2,3,20],[2,5,25]], n = 5
输出：[10,55,45,25,25]
解释：
航班编号        1   2   3   4   5
预订记录 1 ：   10  10
预订记录 2 ：       20  20
预订记录 3 ：       25  25  25  25
总座位数：      10  55  45  25  25
因此，answer = [10,55,45,25,25]

示例 2：

输入：bookings = [[1,2,10],[2,2,15]], n = 2
输出：[10,25]
解释：
航班编号        1   2
预订记录 1 ：   10  10
预订记录 2 ：       15
总座位数：      10  25
因此，answer = [10,25]

提示：

1 <= n <= 2 * 10^4
1 <= bookings.length <= 2 * 10^4
bookings[i].length == 3
1 <= first <= last <= n
1 <= seats <= 104

题目分析

这道题，就是有 n 个航班，告诉你一个订座表，从 first 到 last 号航班，订座数量为 seat。让你求，n 个航班，分别被订的座位。

看上去简单
实际上，确实不难（hhhhh）
很明显，可以暴力求解，两重循环，n平方的时间复杂度。
能行吗？能，就是慢，容易超时
面试如果用暴力法，结果可能是回家等通知

class Solution {
    public int[] corpFlightBookings(int[][] bookings, int n) {
        int[] res = new int[n];
        for (int i = 0; i < bookings.length; i++) {
            int[] book = bookings[i];
            int first = book[0];
            int second = book[1];
            int seat = book[2];

            for(int j = first - 1; j < second; j++) {
                res[j] += seat;
            }
        }
        return res;
    }
}

另一种解法，需要使用差分思想，能使时间复杂度从 O(n^2) 降到 O(n + m)。
n 为 航班数，M 为 bookings 数组的行数。

差分思想是记录差值，使得对某个区间的 O(n) 的增量，优化为 O(1) 的计算。

差分有两个概念：一个是前缀和数组，另一个是差分数组。

原数组：[1,2,2,4]，对应的差分数组是：[1,1,0,2]。
原数组和差分数组的关系是：d [ i ] = a [ i ] - a [ i - 1]，d [ i ] 是差分数组元素。
对差分数组逆运算（第n项=前缀的各项累加），能得到原数组，因此原数组也称为前缀和数组。

好，这就把差分的两个概念和一个性质讲清楚了。

要做这道题，还要了解差分数组的另一个性质：
我们对原数组的 [ i, j ] 区间施加一个增量 k 时，等同于在差分数组上，对 d [ i ] + k，对 d [ j + 1 ] - k。

这样对于区间的修改，就变为了对于两个位置的修改。常用于降低时间复杂度。

因此，对于本题，

1. 定义差分数组 res。
2. 遍历 bookings，计算 res 各项
3. 求 res 的前缀和数组，然后返回前缀和数组

class Solution {
    public int[] corpFlightBookings(int[][] bookings, int n) {
        int[] res = new int[n];
        for (int i = 0; i < bookings.length; i++) {
            int[] book = bookings[i];

            int first = book[0];
            int last = book[1];
            int seat = book[2];

            res[first - 1] += seat;
            if (last < n) {
                res[last] -= seat;
            }
        }
        for (int i = 1; i < n; i++) {
            res[i] += res[i-1];
        }
        return res;
    }
}

差分的时间复杂度： O(n + m)。
n 为 航班数，M 为 bookings 数组的行数

空间复杂度：除了题目要求的数组，未定义额外的数组，因此为 O(1)

对比一下两种方法的结果，差分方法很给力！

知识就是力量，科技是第一生产力

总结一下：

两个概念：

前缀和数组（原数组），差分数组

两个性质：

1、差分序列求前缀和可得原序列

2、将原序列区间 [L,R] 中的元素全部施加增量 k ，可以转化操作为差分序列 L 处 +k，R+1 处 -k

本篇完