Create grumpy-bookstore-owner.py

Author: kamyu104

Python/grumpy-bookstore-owner.py

@@ -0,0 +1,19 @@
+# Time:  O(n)
+# Space: O(1)
+
+class Solution(object):
+    def maxSatisfied(self, customers, grumpy, X):
+        """
+        :type customers: List[int]
+        :type grumpy: List[int]
+        :type X: int
+        :rtype: int
+        """
+        result, max_extra, extra = 0, 0, 0
+        for i in xrange(len(customers)):
+            result += 0 if grumpy[i] else customers[i]
+            extra += customers[i] if grumpy[i] else 0
+            if i >= X:
+                extra -= customers[i-X] if grumpy[i-X] else 0
+            max_extra = max(max_extra, extra)
+        return result + max_extra