[leetcode] 1109. Corporate Flight Bookings

Description

There are n flights, and they are labeled from 1 to n.

We have a list of flight bookings. The i-th booking bookings[i] = [i, j, k] means that we booked k seats from flights labeled i to j inclusive.

Return an array answer of length n, representing the number of seats booked on each flight in order of their label.

Example 1:

Input: bookings = [[1,2,10],[2,3,20],[2,5,25]], n = 5
Output: [10,55,45,25,25]

Constraints:

1. 1 <= bookings.length <= 20000.
2. 1 <= bookings[i][0] <= bookings[i][1] <= n <= 20000.
3. 1 <= bookings[i][2] <= 10000

分析

题目的意思是：给定一个bookings的数组，数组中每个元素包含三个值，i,j,k，表示从第i到第j包含k个座位，现在按照顺序输出每个位置的座位数目。这道题我一开始用了暴力破解，果然超时了，最后参考了一下别人的代码，发现用一个求和数组就可以，我不太明白第一个循环是什么意思，但是我验证过了，却是最后求和得到的是正确的结果，对于例1，这里我把第一个循环得到的res输出输出来：

[10,20+25,-10,-20,0,-25]

然后按照第二个循环求和就能得到答案，我晕了。

代码

class Solution:
    def corpFlightBookings(self, bookings: List[List[int]], n: int) -> List[int]:
        res=[0]*(n+1)
        for i,j,k in bookings:
            res[i-1]+=k
            res[j]-=k
        res.pop()
        for i in range(1,n):
            res[i]+=res[i-1]
        return res        

参考文献

[LeetCode] Python O(n) solution | Top 98% Speed | 9 Lines of Code