航班预订统计(差分算法)

x33g5p2x  于2021-09-19 转载在 其他  
字(0.4k)|赞(0)|评价(0)|浏览(360)

在这里插入图片描述

在这里插入图片描述

数据量大,用暴力方法解决必定超时。

  1. //差分算法
  2. class Solution {
  3. public:
  4. vector<int> corpFlightBookings(vector<vector<int>>& bookings, int n) {
  5. vector<int>ans(n); //差分数组,每个元素存储ans[i]-ans[i]的值
  6. for(auto booking:bookings){
  7. ans[booking[0]-1]+=booking[2];
  8. if(booking[1]<n){
  9. ans[booking[1]]-=booking[2];
  10. }
  11. }
  12. //求出差分数组的前缀和数组即得到原数组
  13. for(int i=1;i<n;i++)
  14. ans[i]=ans[i-1]+ans[i];
  15. return ans;
  16. }
  17. };

相关文章