Create 0525-contiguous-array.py

a93a · web-flow · commit 9ae653432783 · 2024-03-16T11:15:36.000+01:00
diff --git a/python/0525-contiguous-array.py b/python/0525-contiguous-array.py
@@ -0,0 +1,21 @@
+class Solution:
+    def findMaxLength(self, nums: List[int]) -> int:
+        zero, one = 0, 0
+        res = 0
+
+        diff_index = {}
+
+        for i, n in enumerate(nums):
+            if n == 0:
+                zero += 1
+            else:
+                one += 1
+            if one - zero not in diff_index:
+                diff_index[one - zero] = i
+            
+            if one == zero:
+                res = one + zero
+            else:
+                idx = diff_index[one - zero]
+                res = max(res, i - idx)
+        return res
