2024 - Day 08 - Even simpler initial values for generateSequence.

fmmr · Dec 8, 2024 · 898b22f · 898b22f
1 parent 5a0d6fc
commit 898b22f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/main/kotlin/no/rodland/advent_2024/Day08.kt b/src/main/kotlin/no/rodland/advent_2024/Day08.kt
@@ -31,8 +31,8 @@ class Day08(val input: List<String>) : Day<Int, Int, Pair<Array<CharArray>, Map<
 
     private fun Pair<Pos, Pos>.antinodes2(): Set<Pos> {
         return (second - first).let { diff ->
-            val forward = generateSequence(first + diff) { it + diff }.takeWhile { it in grid }
-            val backwards = generateSequence(second - diff) { it - diff }.takeWhile { it in grid }
+            val forward = generateSequence(second) { it + diff }.takeWhile { it in grid }
+            val backwards = generateSequence(first) { it - diff }.takeWhile { it in grid }
             (forward + backwards).toSet()
         }
     }