1109. 航班预订统计 - 力扣(LeetCode) (leetcode-cn.com)
本质上是前缀和加拆分的思想。将n个相同的变化转化为2个(即头和尾)的变化,更加有利于改善时间复杂度。
本题做法即先将数组逆前缀和化,再利用逆前缀和数组的特点,即index位置的值表示的是和数组中index位置和index-1位置的数值之差,对逆前缀和数组进行处理,最后进行前缀和计算得到结果。(此为倒推,方便看懂题解)
正推:
对于一个正常的数组来说,在某一段位置加上某个数对于逆前缀和来说影响的只有头和尾巴+1两个位置。直接变成了每一次O(1)的操作!!!!(目的就是减少时间复杂度)。
class Solution {
public:
vector<int> corpFlightBookings(vector<vector<int>>& bookings, int n) {
vector<int> result(n);
#逆前缀和思想+拆分算法
for(vector<int> i: bookings){
result[i[0]-1]+=i[2];
if (i[1]<n) result[i[1]]-=i[2];
}
#计算前缀和
for(int i=1;i<n;i++) result[i]+=result[i-1];
return result;
}
};