Improved task 3441

jscrdev · Feb 5, 2025 · 5aaa0ce · 5aaa0ce
1 parent 2acb799
commit 5aaa0ce
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/main/kotlin/g3401_3500/s3441_minimum_cost_good_caption/Solution.kt b/src/main/kotlin/g3401_3500/s3441_minimum_cost_good_caption/Solution.kt
@@ -13,11 +13,11 @@ class Solution {
             return ""
         }
         val arr = caption.toCharArray()
-        val prefixCost = Array<IntArray?>(n + 1) { IntArray(26) }
+        val prefixCost = Array<IntArray>(n + 1) { IntArray(26) }
         for (i in 0..<n) {
             val orig = arr[i].code - 'a'.code
             for (c in 0..25) {
-                prefixCost[i + 1]!![c] = prefixCost[i]!![c] + abs((orig - c))
+                prefixCost[i + 1][c] = prefixCost[i][c] + abs((orig - c))
             }
         }
         val dp = IntArray(n + 1)
@@ -37,7 +37,7 @@ class Solution {
                     var bestLetter = 0
                     var bestCost: Int = INF
                     for (c in 0..25) {
-                        val costBlock = prefixCost[i + l]!![c] - prefixCost[i]!![c]
+                        val costBlock = prefixCost[i + l][c] - prefixCost[i][c]
                         if (costBlock < bestCost) {
                             bestCost = costBlock
                             bestLetter = c