作者一开始用的是双指针进行遍历操作最后输入答案。但是这里的最优解其实就是前缀和的逆运算,就是差分。
首先上双指针的做法(因为很简单,所以读者自己体会就行):
class Solution {
public:
vector<int> corpFlightBookings(vector<vector<int>>& bookings, int n) {
int size=bookings.size();
vector<int>cang(n+1);
for(int i=0;i<size;i++){
int left=bookings[i][0];
int right=bookings[i][1];
while(left<=right){
cang[left]+=bookings[i][2];
left++;
}
}
vector<int>res;
for(int j=1;j<=n;j++){
res.push_back(cang[j]);
}
return res;
}
};
好了,下面说一说差分的做法,大家都知道,一旦说要同时改变数组的,而且改变的数值是统一的,这个时候效率最好的做法肯定就是求出它的差分数组,然后将差分数组转换为前缀和数组就可以了。
以前只做过操作一次的差分,但是这次是多次操作改变,其实一样,就依次用一重循环进行加减差分数组就行,不用急着先把差分数组求出来,再化成前缀和,再继续新的差分数组,再求前缀和依次循环。没必要,直接一起在差分数组上操作就行。
注意:可能你看到了区间的左右边都-1了。这是因为在leetcode上你提交的数组是从下标为0开始遍历的,不是从1开始,所以需要都减去1。如果说不在这个leetcode上,也可以不用-1,正常从下标为1开始遍历就行。
class Solution {
public:
vector<int> corpFlightBookings(vector<vector<int>>& bookings, int n) {
int size=bookings.size();
vector<int>chafen(n);
for(int i=0;i<size;i++){
int left=bookings[i][0]-1;
int right=bookings[i][1]-1;
int chazhi=bookings[i][2];
chafen[left]+=chazhi;
if(right<n-1)
chafen[right+1]-=chazhi;
}
for(int i=1;i<n;i++){
chafen[i]+=chafen[i-1];
}
return chafen;
}
};