Medium
Corporate Flight Bookings — Python
Full explanation · Time O(n) · Space O(1)
# Time: O(n)
# Space: O(1)
class Solution(object):
def corpFlightBookings(self, bookings, n):
"""
:type bookings: List[List[int]]
:type n: int
:rtype: List[int]
"""
result = [0]*(n+1)
for i, j, k in bookings:
result[i-1] += k
result[j] -= k
for i in xrange(1, len(result)):
result[i] += result[i-1]
result.pop()
return result