Fix memoization function

edgars-supe · edgars-supe · commit 6faaf64ae0e0 · 2025-11-23T22:58:14.000+02:00
diff --git a/src/main/kotlin/lv/esupe/aoc/utils/Memo.kt b/src/main/kotlin/lv/esupe/aoc/utils/Memo.kt
@@ -2,10 +2,10 @@ package lv.esupe.aoc.utils
 
 fun <T, R> memoize(function: (T) -> R): (T) -> R {
     val cache = mutableMapOf<T, R>()
-    return { t -> cache.computeIfAbsent(t, function) }
+    return { t -> cache.getOrPut(t) { function(t) } }
 }
 
 fun <T, R, V> memoize(function: (T, R) -> V): (T, R) -> V {
     val cache = mutableMapOf<Pair<T, R>, V>()
-    return { t, r -> cache.computeIfAbsent(t to r) { (t, r) -> function(t, r) } }
+    return { t, r -> cache.getOrPut(t to r) { function(t, r) } }
 }

Original file line number	Diff line number	Diff line change
`@@ -2,10 +2,10 @@ package lv.esupe.aoc.utils`
`2`	`2`
`3`	`3`	`fun <T, R> memoize(function: (T) -> R): (T) -> R {`
`4`	`4`	`val cache = mutableMapOf<T, R>()`
`5`		`- return { t -> cache.computeIfAbsent(t, function) }`
	`5`	`+ return { t -> cache.getOrPut(t) { function(t) } }`
`6`	`6`	`}`
`7`	`7`
`8`	`8`	`fun <T, R, V> memoize(function: (T, R) -> V): (T, R) -> V {`
`9`	`9`	`val cache = mutableMapOf<Pair<T, R>, V>()`
`10`		`- return { t, r -> cache.computeIfAbsent(t to r) { (t, r) -> function(t, r) } }`
	`10`	`+ return { t, r -> cache.getOrPut(t to r) { function(t, r) } }`
`11`	`11`	`}`