数据量大,用暴力方法解决必定超时。
//差分算法
class Solution {
public:
vector<int> corpFlightBookings(vector<vector<int>>& bookings, int n) {
vector<int>ans(n); //差分数组,每个元素存储ans[i]-ans[i]的值
for(auto booking:bookings){
ans[booking[0]-1]+=booking[2];
if(booking[1]<n){
ans[booking[1]]-=booking[2];
}
}
//求出差分数组的前缀和数组即得到原数组
for(int i=1;i<n;i++)
ans[i]=ans[i-1]+ans[i];
return ans;
}
};