航班预订统计（LeetCode）

题目

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

        有一份航班预订表 bookings ，表中第 i 条预订记录 bookings[i] = [firsti, lasti, seatsi] 意味着在从 firsti 到 lasti （包含 firsti 和 lasti ）的 每个航班 上预订了 seatsi 个座位。

        请你返回一个长度为 n 的数组 answer，里面的元素是每个航班预定的座位总数。

解题

"""
时间复杂度为 O(n + k)，其中 n 是航班的数量，k 是预订记录的数量
"""


def corpFlightBookings(bookings, n):
    # 初始化差分数组
    diff = [0] * (n + 1)

    # 处理每个预订记录
    for first, last, seats in bookings:
        diff[first - 1] += seats
        if last < n:
            diff[last] -= seats

    # 根据差分数组计算每个航班的座位总数
    answer = [0] * n
    answer[0] = diff[0]
    for i in range(1, n):
        answer[i] = answer[i - 1] + diff[i]

    return answer


bookings = [[1, 2, 10], [2, 3, 20], [2, 5, 25]]
n = 5
print(corpFlightBookings(bookings, n))  # [10, 55, 45, 25, 25]