Leetcode 1109. Corporate Flight Bookings [Python]

Case_CaiNiao

于 2021-12-28 00:30:16 发布

阅读量287

点赞数

分类专栏： Leetcode学习记录文章标签： leetcode python 算法

本文链接：https://blog.csdn.net/sinat_30403031/article/details/122183471

版权

Leetcode学习记录专栏收录该内容

268 篇文章

订阅专栏

该篇博客介绍了如何利用前缀和技巧来处理航班预订的累积效果。通过遍历航班预订列表，更新每个航班的座位增减，然后计算前缀和，最终得到每个航班的座位占用情况。这种方法高效地解决了数据累加的问题。

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

前缀和，在一段空array上，遇到start的航班号的index，+相应的reserve的座位数，这个数字在将会在前缀和操作时一直往后面的航班上加，加到end+1处则停止，所以，在end+1的index位置-seats数。随后求array前缀和就可以了。

class Solution:
    def corpFlightBookings(self, bookings: List[List[int]], n: int) -> List[int]:
        res = [0]*100010
        for start, end, seats in bookings:
            res[start] += seats
            res[end+1] -= seats
        res = res[1:]
            
        prefix = [0]
        for i in range(len(res)):
            prefix.append(prefix[-1] + res[i])
        return prefix[1:n+1]