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]

更新於: 30-4 月 -2020

145 次瀏覽

啟動你的職業生涯

透過完成課程獲得認證

開始
廣告