Python 中的公司航班预订
假设我们有 n 个航班,它们从 1 到 n 标记。我们有一个航班预订清单。第 i 个预订表示使用 bookings[i] = [i, j, k],这意味着我们从标记为 i 到 j 的航班(包括 j)预订了 k 个座位。找到一个长度为 n 的答案数组,按其标签的顺序显示在每个航班上预订的座位数。因此,如果输入类似于 [[1,2,10],[2,3,20],[2,5,25]] 且 n = 5,则输出将为 [10, 55, 45, 25, 25]。
要解决这个问题,我们将按照以下步骤操作:
- res := 制作一个大小为 n 的数组,并用 0 填充它
- 对于 bookings 中的每个条目 i
- res[i[0] - 1] := res[i[0] - 1] + i[2]
- 如果 i[1] < n,那么 res[i[1]] := res[i[1]] – i[2]
- 对于 i 在范围 1 到 n – 1
- res[i] := res[i] + res[i - 1]
- 返回 res
让我们看看以下实现,以更好地理解它:
示例
class Solution(object): def corpFlightBookings(self, bookings, n): res = [0 for i in range(n)] for i in bookings: res[i[0]-1]+=i[2] if(i[1]<n): res[i[1]]-=i[2] for i in range(1,n): res[i]+=res[i-1] return res ob = Solution() print(ob.corpFlightBookings([[1,2,10],[2,3,20],[2,5,25]],5))
输入
[[1,2,10],[2,3,20],[2,5,25]] 5
输出
[10, 55, 45, 25, 25]
广告