Create max-consecutive-ones-iii.py

Author: kamyu104

Python/max-consecutive-ones-iii.py

@@ -0,0 +1,18 @@
+# Time:  O(n)
+# Space: O(1)
+
+class Solution(object):
+    def longestOnes(self, A, K):
+        """
+        :type A: List[int]
+        :type K: int
+        :rtype: int
+        """
+        result, i = 0, 0
+        for j in xrange(len(A)):
+            K -= int(A[j] == 0)
+            while K < 0:
+                K += int(A[i] == 0)
+                i += 1
+            result = max(result, j-i+1)
+        return result