Improved task 3213

jscrdev · Jul 15, 2024 · c3ac45a · c3ac45a
1 parent 8f3e250
commit c3ac45a
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/main/java/g3201_3300/s3213_construct_string_with_minimum_cost/Solution.java b/src/main/java/g3201_3300/s3213_construct_string_with_minimum_cost/Solution.java
@@ -86,7 +86,9 @@ public int minimumCost(String target, String[] words, int[] costs) {
         dp[0] = 0;
         ACAutomaton.Node node = root;
         for (int i = 1; i < dp.length; i++) {
-            node = ac.go(node, target.charAt(i - 1));
+            if (node != null) {
+                node = ac.go(node, target.charAt(i - 1));
+            }
             for (ACAutomaton.Node temp = node;
                     temp != null && temp != root;
                     temp = ac.getOutput(temp)) {