LeetCode 力扣 1109 航班预订统计 差分 前缀和 彻底搞懂差分

题目链接：航班预订统计
题目描述：

初始给定一个元素全为$0$的数组，给定很多操作，操作描述为$l,r,val$表示将索引区间$[l,r]$上的数字增加$val$，返回所有操作之后数组。

题解：

这道题目当然可以用线段树、树状数组等来解决，这里介绍如何用差分来解决。这道题没有中间的查询，只有最后一次查询这才有了差分的可能性。
什么是差分呢？差分通常是指做差，而这道题目中，我们实际上用到了差分和前缀和两个技巧：我们设差分数组为$c[i]=a[i]-a[i-1]$（其中$a[i]$为原始数组，特别的$c[0]=a[0]$）。
这样可以得到一个显然的性质：$a[i]=a[i]-a[i-1]+a[i-1]-a[i-2]+...+a[1]-a[0]+a[0]=c[i]+c[i-1]+...+c[1]+c[0]$就是说我们对$c$数组做一次前缀和就能得到$a$数组，这样做有什么好处呢？
我们来看如果对原数组进行区间增加操作，差分数组会发生什么操作，例如我们将$[l,r]$的数全部增加$x$那么原数组中这一部分的数将变为：$a[l]+x,a[l+1]+x,...,a[r]+x$，此时我们来看新的差分数组$c^{\prime }$有：
$c^{\prime }[l]=a[l]+x-a[l-1]=c[l]+x$
$c^{\prime }[l+1]=a[l+1]+x-(a[l]+x)=c[l+1]$
…
$c^{\prime }[r]=a[r]+x-(a[r-1]+x)=c[r]$
$c^{\prime }[r+1]=a[r+1]-(a[r]+x)=c[r+1]-x$
可以看到对于原数组进行区间增加的操作后，差分数组中只有$l,r+1$两处位置的值发生了改变，这就意味着，对于原数组的区间增加操作可以在差分数组上以时间复杂度$O(1)$来实现，在所有增加操作执行完之后，再对差分数组求一次前缀和，就能够得到更新后的原数组。在本题中由于初始的数组元素全为$0$，那么差分数组也就全为$0$。

代码：

class Solution {
public:
    vector<int> corpFlightBookings(vector<vector<int>>& bookings, int n) {
        vector<int> ans(n, 0);
        for (auto op : bookings) {
            int l = op[0] - 1, r = op[1] - 1, val = op[2];
            ans[l] += op[2];
            if (r + 1 < n) { ans[r + 1] -= op[2]; }
        }
        for (int i = 1; i < n; i++) { ans[i] += ans[i - 1]; }
        return ans;
    }
};