LeetCode1109. 航班预订统计 (前缀和 + 差分)

GSX_M

已于 2022-03-17 19:29:20 修改

阅读量253

点赞数

分类专栏：力扣刷题文章标签：算法前缀和差分

于 2022-03-17 19:27:34 首次发布

本文链接：https://blog.csdn.net/m0_52169086/article/details/123558207

版权

力扣刷题专栏收录该内容

64 篇文章 0 订阅

订阅专栏

本文介绍了一种解决力扣平台上的航班预订问题的方法，通过使用前缀和与差分策略优化时间复杂度，从原来的O(m*n)降低到O(n)，其中n是数组长度，m是预定记录数。代码示例展示了如何利用C++实现这一优化，通过遍历预定记录并更新座位占用状态，最终返回每个航班的座位占用情况。

摘要由CSDN通过智能技术生成

力扣

解题思路:

1.创建一个vector ，遍历bookings，在区间vector 区间[ first , last ] 都加上 seats，

但是超时了。时间复杂度O(m*n) ,其中 n为要求的数组长度，m 为预定记录的数量.

2.前缀和+差分 ( 具体的思想讲解-> B站星锤月朦胧 )

class Solution {
public:
    vector<int> corpFlightBookings(vector<vector<int>>& bookings, int n) 
    {
       vector<int> ret(n, 0);

       for(vector<int>& e : bookings)
       {
          ret[e[0]-1] += e[2];
          if(e[1] < n)
          {
              ret[e[1]] -= e[2];
          }
       }

       for(int i = 0 ; i < n-1 ;++i)
       {
           ret[i+1] += ret[i];
       }

       return ret;
    }
};