力扣 1109. 航班预订统计 差分

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

思路：最简单的想法，遍历每个区间修改座位数，那么总复杂度可能达到$O(n^2)$，大概率会超时。如果你学过线段树/树状数组的话，也可以用$O(nlgn)$的复杂度解决这个问题，就是一个区间修改和单点查询嘛。但是我们有更加巧妙的复杂度为$O(n)$的差分算法。假设最终所求数组为$ans$，我们可以计算其差分数组：$dif{f}_0=a_0,dif{f}_i=a_i-a_{i-1}$。显然有$a_i={\sum }_{j=0}^{i}dif{f}_j$。那么对$a[l...r]$增加$v$的操作就相当于对$dif{f}_l$做$+v$操作，对$dir{r}_{r+1}$做$-v$操作。

class Solution {
public:
    vector<int> corpFlightBookings(vector<vector<int>>& bookings, int n) {
        vector<int> cnt(n+2);
        for(const vector<int>& booking:bookings)
            cnt[booking[0]]+=booking[2],cnt[booking[1]+1]-=booking[2];
        vector<int> ans(n);
        int curSeats=0;
        for(int i=1;i<=n;i++)
        {
            curSeats+=cnt[i];
            ans[i-1]=curSeats;
        }
        return ans;
    }
};