题目:
题目地址:here
这里有
n
n
n个航班,它们分别从
1
1
1到
n
n
n 进行编号。
有一份航班预订表 bookings ,表中第 i i i条预订记录 bookings[ i i i] = [first i _i i, last i _i i, seats i _i i] 意味着在从 first i _i i 到 last i _i i (包含 first i _i i 和 last i _i i )的 每个航班 上预订了 seats i _i i 个座位。
请你返回一个长度为 n n n 的数组 answer,其中 answer[ i i i] 是航班 i 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 * 104
1 <= bookings.length <= 2 * 104
bookings[i].length == 3
1 <= firsti <= lasti <= n
1 <= seatsi <= 104
题解:
解题思路来自博主:青格儿
- 换一种思路理解题意,将问题转换为:某公交车共有 n 站,第 i 条记录 bookings[i] = [i, j, k] 表示在 i 站上车 k 人,乘坐到 j 站,在 j+1 站下车,需要按照车站顺序返回每一站车上的人数
- 根据 1 的思路,定义 counter[] 数组记录每站的人数变化,counter[i] 表示第 i+1 站。遍历 bookings[]:bookings[i] = [i, j, k] 表示在 i 站增加 k 人即 counters[i-1] += k,在 j+1 站减少 k 人即 counters[j] -= k
- 遍历(整理)counter[] 数组,得到每站总人数: 每站的人数为前一站人数加上当前人数变化 counters[i] += counters[i - 1]
其实解题思路就是前缀和:后面的值是前面值的累加。
class Solution:
def corpFlightBookings(self, bookings: List[List[int]], n: int) -> List[int]:
dp = [0] * n
# 解法一 超时
# for e in bookings:
# for i in range(e[0], e[1]+1):
# dp[i-1] += e[2]
# 解法二
for e in bookings:
dp[e[0]-1] += e[2]
if e[1] < n:
dp[e[1]] -= e[2]
for i in range(1, n):
dp[i] += dp[i-1]
return dp