LeetCode 1109中等数组
航班预订统计 图解题解
这道题到底在问什么
请你返回一个长度为 n 的数组 answer ,里面的元素是每个航班预定的座位总数。
- bookings,n
- [[1,2,10],[2,3,20],[2,5,25]],5
- 输出
- [10,55,45,25,25]
先想最直接的笨办法
数组/字符串状态条跟着代码走:推进语句是:cur += diff[i]。处理过的部分不再重新枚举。(动画第 10 步)
最优解:一步一步想明白
- 3下面 9 步动画会按主解代码推进,而不是泛泛讲题型。
- 4先读清 航班预订统计 的输入输出数组/字符串状态条跟着代码走:先把示例输入映射到代码参数:def corpFlightBookings(self, bookings, n):。
- 5diff = [0] * (n + 1)数组/字符串状态条跟着代码走:开局只立住必要变量:diff = [0] * (n + 1)。
- 6for first, last, seats in bookings:数组/字符串状态条跟着代码走:主流程从这里开始:for first, last, seats in bookings:。
- 7根据题意分情况数组/字符串状态条跟着代码走:题目条件落到这一行:满足条件就进入对应分支。
- 8diff[first - 1] += seats数组/字符串状态条跟着代码走:对应代码:diff[first - 1] += seats。这一行决定当前轮对答案有什么贡献。
- 9用差分数组数组/字符串状态条跟着代码走:边界跟着代码看:diff[last] -= seats。
- 10cur += diff[i]数组/字符串状态条跟着代码走:推进语句是:cur += diff[i]。处理过的部分不再重新枚举。
- 11return ans数组/字符串状态条跟着代码走:到这里,diff 已经能表达题目要求。
- 12return:return ans数组/字符串状态条跟着代码走:最后检查返回形态:返回值、原地修改或设计类状态,要和 LeetCode 判题方式一致。
- 15记住这题的代码骨架:题意约束先落到状态变量,再用循环或递归维护它。
⚠️ 容易写错的地方
✗ 错:每个区间逐个座位加
✓ 对:用差分数组
区间很多时会超时
✗ 错:下标没转成 0-based
✓ 对:first-1 和 last 处理
航班编号从 1 开始
完整代码(Python)
Python
class Solution:
def corpFlightBookings(self, bookings, n):
diff = [0] * (n + 1)
for first, last, seats in bookings:
diff[first - 1] += seats
diff[last] -= seats
ans = []
cur = 0
for i in range(n):
cur += diff[i]
ans.append(cur)
return ans复杂度
时间复杂度
O(n+m)
处理预订再前缀还原
空间复杂度
O(n)
差分数组
看不够?换成动画再走一遍
上面的推演每一步都对应一帧动画。点开交互动画版,能一步步看着 航班预订统计 的数据怎么变、指针怎么走,还能切 Python / Java / C++ 跟着练。
面试官可能追问
这道题为什么用「数组」,换最直接的暴力解会差在哪?+
数组抓住了本题的结构特征,把暴力解里重复的工作省掉;暴力解通常要多嵌套一层枚举,数据一大就超时。具体对比见上文「暴力解及其卡点」与「最优解逐步推演」两节。
时间复杂度为什么是 undefined?怎么推出来的?+
按上文复杂度小节的推导,时间复杂度为 undefined。
本平台为独立第三方培训机构,与华为技术有限公司无任何关联;课程的服务内容与权益以购买协议为准,学习效果因个人情况而异。「华为 OD」「华为可信」等仅为对岗位与考试方向的客观描述,相关商标归各自权利人所有。