Create 0525-contiguous-array.kt

a93a · web-flow · commit bb0af3d25c8d · 2024-03-16T11:07:33.000+01:00
diff --git a/kotlin/0525-contiguous-array.kt b/kotlin/0525-contiguous-array.kt
@@ -0,0 +1,27 @@
+class Solution {
+    fun findMaxLength(nums: IntArray): Int {
+        var zero = 0
+        var one = 0
+        var res = 0
+
+        val diffIndex = HashMap<Int, Int> ()
+
+        for ((i, n) in nums.withIndex()) {
+            if (n == 0)
+                zero++
+            else
+                one++
+            if (one - zero !in diffIndex)
+                diffIndex[one - zero] = i
+
+            if (one == zero) {
+                res = one + zero
+            } else {
+                val idx = diffIndex[one - zero] ?: 0
+                res = maxOf(res, i - idx)
+            }
+        }
+
+        return res
+    }
+}
