Create corporate-flight-bookings.py

Author: kamyu104

Python/corporate-flight-bookings.py

@@ -0,0 +1,18 @@
+# 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