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

---

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