题目来源力扣:
这里有 n 个航班,它们分别从 1 到 n 进行编号。
有一份航班预订表 bookings ,表中第 i 条预订记录 bookings[i] = [firsti, lasti, seatsi] 意味着在从 firsti 到 lasti (包含 firsti 和 lasti )的 每个航班 上预订了 seatsi 个座位。
请你返回一个长度为 n 的数组 answer,其中 answer[i] 是航班 i 上预订的座位总数。
差分数组法:
差分数组的性质是,当我们希望对原数组的某一个区间 [l,r] 施加一个增量inc 时,差分数组d对应的改变是:d[l]增加 inc,d[r+1] 减少 inc。这样对于区间的修改就变为了对于两个位置的修改。并且这种修改是可以叠加的,即当我们多次对原数组的不同区间施加不同的增量,我们只要按规则修改差分数组即可。引用自:LeetCode-Solution
class Solution {
public:
vector<int> corpFlightBookings(vector<vector<int>>& bookings, int n) {
vector<int> count(n);
for(auto &book:bookings){ //遍历区间,带不带引用符号都可以
count[book[0] - 1] += book[2]; //原n从1开始 vector从0开始 所以有book[0]-1
if(book[1] < n){ //等于n的时候超出范围 不考虑
count[book[1] ] -= book[2];
}
}
for(int i = 1; i < n; i++){
count[i] += count[i-1];
}
return count;
}
};