- 航班预订统计
这里有 n 个航班,它们分别从 1 到 n 进行编号。
有一份航班预订表 bookings ,表中第 i 条预订记录 bookings[i] = [firsti, lasti, seatsi] 意味着在从 firsti 到 lasti (包含 firsti 和 lasti )的 每个航班 上预订了 seatsi 个座位。
请你返回一个长度为 n 的数组 answer,里面的元素是每个航班预定的座位总数。
本题用到的解题思想是差分数组。差分数组是前缀和的逆运算。
presum[i]=presum[i-1]+nums[i]
d[i]=nums[i]-nums[i-1]
求差分数组的前缀和,结果即原数组。
在区间[l, r]上有增量delta,反映在差分数组上,d[l]+=delta,d[r+1]-=delta,如果r=n-1,则d[n-1]不需要变化。
本题可以先初始化一个差分数组,对于每一条预订记录,在差分数组下标为first和last的元素加减变量。最后对差分数组求前缀和,即得到结果。
class Solution {
public:
vector<int> corpFlightBookings(vector<vector<int>>& bookings, int n) {
vector<int>difference(n);
for(int i=0;i<bookings.size();i++){
int first=bookings[i][0],last=bookings[i][1],seat=bookings[i][2];
difference[first-1]+=seat;
if(last!=n)
difference[last]-=seat;
}
for(int i=1;i<n;i++){
difference[i]+=difference[i-1];
}
return difference;
}
};