题目来源于leetcode,解法和思路仅代表个人观点。传送门。
难度:中等
时间:-
TAG:差分数组
题目
这里有 n 个航班,它们分别从 1 到 n 进行编号。
有一份航班预订表 bookings ,表中第 i 条预订记录 bookings[i] = [firsti, lasti, seatsi] 意味着在从 firsti 到 lasti (包含 firsti 和 lasti )的 每个航班 上预订了 seatsi 个座位。
请你返回一个长度为 n 的数组 answer,其中 answer[i] 是航班 i 上预订的座位总数。
示例 1:
输入:bookings = [[1,2,10],[2,3,20],[2,5,25]], n = 5
输出:[10,55,45,25,25]
解释:
航班编号 1 2 3 4 5
预订记录 1 : 10 10
预订记录 2 : 20 20
预订记录 3 : 25 25 25 25
总座位数: 10 55 45 25 25
因此,answer = [10,55,45,25,25]
示例 2:
输入:bookings = [[1,2,10],[2,2,15]], n = 2
输出:[10,25]
解释:
航班编号 1 2
预订记录 1 : 10 10
预订记录 2 : 15
总座位数: 10 25
因此,answer = [10,25]
提示:
1 <= n <= 2 * 104
1 <= bookings.length <= 2 * 104
bookings[i].length== 3
1 <= firsti <= lasti <= n
1 <= seatsi <= 104
思路
如果生成一个二维数组保存bookings,在累加。那么肯定会超时。
我们需要一个方法,维护一段区间,使得在某段区间的修改,变得容易。并且能够在较短时间内获得,区间某个位置的值。
这个方法就是差分数组。
差分数组
差分数组的定义?
对于数组 [1,2,2,4],其差分数组为 [1,1,0,2],差分数组的第 i i i个数即为原数组的第 i − 1 i-1 i−1个元素和第 i i i个元素的差值。
差分数组的特点?
- 对差分数组求前缀和即可得到原数组。(前缀和可以原地修改原数组)。
- 当我们想对原数组的某个区间 [ l , r ] [l,r] [l,r]施加一个增量 i n c inc inc时,差分数组 d d d对应的改变是: d [ l ] d[l] d[l]增加 i n c inc inc, d [ r + 1 ] d[r+1] d[r+1]减少 i n c inc inc。这样对于区间的修改就变为了对于两个位置的修改。
代码
class Solution {
public:
vector<int> corpFlightBookings(vector<vector<int>>& bookings, int n) {
//区间查询
vector<int> ans(n+1);
for(auto &booking:bookings){
int first = booking[0];
int last = booking[1];
int seats = booking[2];
ans[first-1] += seats;
ans[last] -= seats;
}
//计算前缀和
for(int i=1;i<n;i++){
ans[i] += ans[i-1];
}
ans.pop_back();
return ans;
}
};
算法复杂度
时间复杂度: O(n+m)。n为航班数量,m为bookings的长度。
空间复杂度: O(n)。n为差分数组ans长度。