LeetCode 刷题系列 -- 1109. 航班预订统计

这里有 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 * 104

• 1 <= bookings.length <= 2 * 10^4

• bookings[i].length == 3

• 1 <= firsti <= lasti <= n

• 1 <= seatsi <= 10^4

1109. 航班预订统计 - 力扣（Leetcode）

思路：小而美的算法技巧：差分数组 :: labuladong的算法小抄

此种题目常用 差分数组求解。
定义 diff 为差分数组，diff[ i ] = result[ i ] - result[ i - 1]
1. 初始化 diff数组，(i, j, val) 表示对 nums[ i ], ... , nums[ j ] 中的每个元素分别加上 val
1）diff[ i ] += val
2) diff[ j +1] -= val , 因为只对 nums[ i ], ... , nums[ j ] 中的每个元素分别加上 val，而 nums[ j + 1] 及其以后的元素不再加上 val
2. 更新结果 result

c++

class Solution {
public:
    vector<int> diff; // 差分数组，diff[i] = result[i] - result[i-1]

    void initDiff(vector<int>& diff, int i, int j, int val) {
        diff[i-1] += val;
        if(j < diff.size()) {
            diff[j] -= val;
        }
    }

    vector<int> corpFlightBookings(vector<vector<int>>& bookings, int n) {
        vector<int> result(n, 0); // result[i] 表示 第 i+1 个编号的航班
        diff = vector<int>(n, 0);

        for(auto& vec:bookings) {
            int first = vec[0];
            int last = vec[1];
            int seat = vec[2];
            initDiff(diff, first, last, seat);
        }

        result[0] = diff[0];

        for(int i=1; i<n; i++) {
            result[i] = diff[i] + result[i-1];
        }

        return result;
    }
};