From d5d8cb1748ac57c8d42916ca985ca46901de9cc6 Mon Sep 17 00:00:00 2001
From: Valentyn Kolesnikov <javadev75@gmail.com>
Date: Wed, 5 Feb 2025 07:49:13 +0200
Subject: [PATCH] Improved task 3441

---
 .../g3401_3500/s3441_minimum_cost_good_caption/Solution.kt  | 6 +++---
 1 file changed, 3 insertions(+), 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
index f359986b..a265d2ab 100644
--- 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