力扣 1109. 航班预订统计

题目来源：https://leetcode-cn.com/problems/corporate-flight-bookings/

大致题意：
给定一个二维数组 bookings 和一个整数 n，其中 bookings 每一维都只有三个元素，代表起始站点和终点站点，以及这个区间内航班数量；n 表示航站数量。
返回一个一维数组，代表每个站点所有的航班数量。

思路

暴力遍历

1. 遍历 bookings 的每一维，得到起始航站编号和航班数量
2. 遍历起始区间的所有航站，每个航站加上本轮航班数量

代码：

public int[] copyFlightBookings(int[][] bookings, int n) {
        int[] flightsNum = new int[n];
        for (int[] booking : bookings) {
            // 区间增值
            int begin = booking[0] - 1;
            int end = booking[1] - 1;
            int num = booking[2];
            for (int i = begin; i <= end; i++) {
                flightsNum[i] += num;
            }
        }
        return flightsNum;
    }

时间复杂度 O(n^2)

差分

差分数组：除了首位元素外，对于索引 i 来说，差分位置 i 的数组元素为 – 原数组索引 i 处元素 减去 原数组索引 i-1 处元素。假定原数组为[1, 2, 3, 4]，那么其差分数组为[1, 1, 1, 1]。而若对差分数组求前缀和，又会得到原数组，如[1, 1, 1, 1]求前缀和，得[1, 2, 3, 4]

在有了差分数组的概念后，若我们对区间 [l, r] 的元素加上 num，那么差分数组索引 l 位置的元素值将会加 num（l 位置元素增加 num，l-1 位置不变，差值增加 num），索引 r+1 位置差分元素将会减去 num（因为 r 和 r-1 都增加了 num，那么差也不变，于是 r 位置差分元素值不变。但是 r+1 并未增加，因而 差将会减去 num）。

1. 遍历 bookings 数组，得到起始航站编号和航班数量 num
2. 对于起始位置，使该索引处差分数组元素加上 num；对于结束位置，其后一个位置的差分数组元素 减去 num
3. 遍历差分数组求前缀和，即得到原数组

时间复杂度 O(n + m)，m 指 bookings 第一维的大小
代码：

public int[] copyFlightBookings(int[][] bookings, int n) {
        int[] diffNum = new int[n];
        // 更新差分数组
        for (int[] booking : bookings) {
        	// 索引从 0 开始，需减 1
            diffNum[booking[0] - 1] += booking[2];
            // 最后一位不用管
            // 因为最后一位（航站编号n+1，越界）的航班数量为0
            // 那么其前缀和也为0，况且越界
            if (booking[1] == n)    continue;
            diffNum[booking[1]] -= booking[2];
        }
        // 求前缀和
        for (int i = 1; i < n; i++) {
            diffNum[i] += diffNum[i - 1];
        }

        return diffNum;
    }