【leetcode刷题笔记】5118. 航班预订统计

最新推荐文章于 2023-01-16 19:48:15 发布

白月光soul

最新推荐文章于 2023-01-16 19:48:15 发布

阅读量725

点赞数 1

CC 4.0 BY-SA版权

分类专栏： c++ leetcode刷题笔记文章标签： leetcode

本文链接：https://blog.csdn.net/burningdzb/article/details/95031518

c++ 同时被 2 个专栏收录

14 篇文章

订阅专栏

leetcode刷题笔记

8 篇文章

订阅专栏

本文针对LeetCode上的航班预订问题，提出了一种利用前缀和数组的高效算法解决方案。通过仅处理区间的起始和结束点，避免了直接暴力计算可能带来的超时问题，实现了快速准确地计算每个航班上预订的座位数。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目链接：https://leetcode-cn.com/contest/weekly-contest-144/problems/corporate-flight-bookings/

1.题目

这里有 n 个航班，它们分别从 1 到 n 进行编号。

我们这儿有一份航班预订表，表中第 i 条预订记录 bookings[i] = [i, j, k] 意味着我们在从 i 到 j 的每个航班上预订了 k 个座位。

请你返回一个长度为 n 的数组 answer，按航班编号顺序返回每个航班上预订的座位数。

2.思路

直接暴力计算会超时，所以用前缀和数组计算答案，只需要处理区间端点，让区间起点加预定数量，终点下一位减预定数量，最后让ans[i] = ans[i] + ans[i-1] (i>=1)

3.代码

class Solution
{
public:
    vector<int> corpFlightBookings(vector<vector<int>>& bookings, int n) 
    {
        vector<int>ans(n);
        for(int i = 0; i < bookings.size(); i++)
        {
            int l =  bookings[i][0];
            int r =  bookings[i][1];
            int c =  bookings[i][2];
            ans[l-1] += c;
            if(r<n)
                ans[r] -= c;
        }
        for(int i = 1; i < n; i++)
        {
            ans[i]=ans[i-1]+ans[i];
        }
        
        return ans;
    }
};