From ddd34f68f59d62a48dffcc76d51eeae6dfd5c494 Mon Sep 17 00:00:00 2001
From: Andrew Glaze <andrew@candy123.moe>
Date: Sat, 7 Dec 2024 15:06:04 -0500
Subject: [PATCH] day05 update

---
 app/src/main/kotlin/day05/Day05.kt | 30 ++++++------------------------
 1 file changed, 6 insertions(+), 24 deletions(-)

diff --git a/app/src/main/kotlin/day05/Day05.kt b/app/src/main/kotlin/day05/Day05.kt
index b6eec22..9c0862c 100644
--- a/app/src/main/kotlin/day05/Day05.kt
+++ b/app/src/main/kotlin/day05/Day05.kt
@@ -33,33 +33,15 @@ class Day05 {
         return true
     }
 
-    fun fixOrder(rules: HashMap<Int, List<Int>>, update: List<Int>): List<Int> {
-        var modUpdate = update.toMutableList()
-        for (i in 0..<update.count()) {
-            val curPage = modUpdate[i]
-            val curRule = rules[curPage]
-            if (curRule == null) { continue }
-            for (pre in 0..<i) {
-                if (curRule.contains(modUpdate[pre])) {
-                    modUpdate[i] = modUpdate[pre]
-                    modUpdate[pre] = curPage
-                    break
-                }
-            }
-        }
-        // hate this but im lazy
-        if (!checkOrder(rules, modUpdate)) {
-            modUpdate = fixOrder(rules, modUpdate).toMutableList()
-        }
-        return modUpdate
-    }
-
     fun part2(input: String): Int {
         val (rules, updates) = parse(input)
         return updates.filter {update -> !checkOrder(rules, update) }
-            .map { update -> fixOrder(rules, update) }
-            .map { update -> update[update.count() / 2] }
-            .sum()
+        .flatMap { update ->
+            update.filter { curPage ->
+                rules[curPage]?.filter { update.contains(it) }?.count() == update.count() / 2
+            }
+        }
+        .sum()
     }
 }