fix: builder

borgoat · May 5, 2022 · 4c49b71 · 4c49b71
1 parent e59e85e
commit 4c49b71
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/.idea/misc.xml b/.idea/misc.xml
diff --git a/src/main/kotlin/com/github/borgoat/markovaldo/MarkovChain.kt b/src/main/kotlin/com/github/borgoat/markovaldo/MarkovChain.kt
@@ -28,12 +28,10 @@ class MarkovChain(
       val tokenState = ArrayDeque<Token>()
       tokens.forEach {
         if (tokenState.isNotEmpty()) {
-          val innerMap = probs.getOrPut(tokenState) { mutableMapOf() }
+          val innerMap = probs.getOrPut(tokenState.takeLast(order)) { mutableMapOf() }
           innerMap.getOrPut(it) { AtomicLong(0) }.incrementAndGet()
         }
-
         tokenState.add(it)
-        if (tokenState.count() > order) tokenState.removeFirst()
       }
     }