LeetCode 赛题（航班预订统计）

5118. 航班预订统计（LeetCode 第 144 场周赛）

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

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

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

 

示例：

输入：bookings = [[1,2,10],[2,3,20],[2,5,25]], n = 5
输出：[10,55,45,25,25]

 

提示：

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

   

•  

• 有时提交能通过，又是又通不过，得看运气

• import math

  class Solution:
    def corpFlightBookings(self, bookings, n: int):
        g = 75     # 组大小
        m = n//g    # 组个数
        r = n - m*g  # 最后一组的个数
        s = [0]*m   # 每组整体预订数量
        a = [0]*n   # 组外，每个航班预订数量
        for i in bookings:
            b0 = math.ceil(i[0]/g)
            b1 = i[1]//g 
            if b1 < b0:
                b1 = b0
                
            for j in range(b0, b1):
                s[j] += i[2] # 一条记录中，从i[0]到i[1]的航班中，计算各组预订数量
            
            for j in range(i[0]-1, min(b0*g, i[1])): # 一条记录中，计算不够一组的左边零头航班预订数量
                a[j] += i[2] #--

              for j in range(max(b0, b1)*g, i[1]): # 一条记录中，计算不够一组时，右左边零头航班预订数量
                a[j] += i[2]
        
        b = []
        for i in range(m):
            b.extend([s[i]]*g) # 加速的要点，避免依次刷新各航班数量
        
        b.extend([0]*r)
        
        for i in range(n): # 添加零头
            b[i] += a[i]
        
        return b