力扣每日一题2021.08.31航班预订统计

最新推荐文章于 2022-01-28 16:16:15 发布

wcy1034036507

最新推荐文章于 2022-01-28 16:16:15 发布

阅读量157

点赞数 1

分类专栏： leetcode每日一题文章标签： leetcode

本文链接：https://blog.csdn.net/wcy1034036507/article/details/120010409

版权

leetcode每日一题专栏收录该内容

304 篇文章 6 订阅

订阅专栏

本文介绍了一种利用差分数组高效解决航班预订统计的方法。通过遍历预订记录，更新差分数组并在每个航班处累加，可以得到每个航班的预订座位总数。示例展示了如何应用此方法来处理不同预订场景，并提供了相应的代码实现。

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

文章目录

1109.航班预订统计

1109.航班预订统计

题目描述

这里有n个航班，分别从1到n进行编号。
有一份航班预订表bookings，表中第i条预订记录bookings[i] = [firsti, lasti, seatsi]意味着在从firsti到lasti（包含firsti和lasti）的每个航班上预定了seatsi个座位。
请你返回一个长度为n的数组answer，其中answer[i]是航班i上预订的座位总数。

示例1

输入：bookings = [[1, 2, 10], [2, 3, 20], [2, 5, 25]], n = 5
输出：[10, 55, 45, 25, 25]
解释：
航班编号 1 2 3 4 5
预订记录1 10 10
预订记录2 20 20
预订记录3 25 25 25 25
总座位数 10 55 45 25 25
因此，answer = [10, 55, 45, 25, 25]

示例2

输入：bookings = [[1, 2, 10], [2, 2, 15]], n = 2
输出：[10, 25]
解释：
航变编号 1 2
预订记录1 10 10
预订记录2 15
总座位数 10 25
因此，answer = [10, 25]

提示

1 <= n <= 2*10⁴
1 <= bookings.length <= 2*10⁴
bookings[i].length == 3
1 <= firsti <= lasti <= n
1 <= seatsi <= 10⁴

思路：差分数组

可以使用差分数组统计出每个航班最终是多少座位。

class Solution:
    def corpFlightBookings(self, bookings: List[List[int]], n: int) -> List[int]:
        diff = [0] * (n+1)
        for first, last, seats in bookings:
            diff[first-1] += seats
            diff[last] -= seats
        cur = 0
        ans = []
        for i in range(n):
            cur += diff[i]
            ans.append(cur)
        return ans