From 583e10b4e086ace856ecdcb3ea9be9d1363ff28e Mon Sep 17 00:00:00 2001
From: Armin Friedl <dev@friedl.net>
Date: Thu, 5 Dec 2024 06:58:29 +0100
Subject: [PATCH] Day05
---
day05.py | 52 ++++++++++++++++++++++++++++++++++++++++++++++++++++
1 file changed, 52 insertions(+)
create mode 100644 day05.py
diff --git a/day05.py b/day05.py
new file mode 100644
index 0000000..25efb27
--- /dev/null
+++ b/day05.py
@@ -0,0 +1,52 @@
+#!/usr/bin/env python3
+
+from functools import cmp_to_key
+
+
+o = {}
+p = {}
+i = []
+
+with open("day05") as f:
+ l = f.readline()
+ while l != '\n':
+ a,b = map(int, l.strip().split('|'))
+
+ if a in o: o[a].add(b)
+ else: o[a] = {b}
+
+ if b in p: p[b].add(a)
+ else: p[b] = {a}
+
+ l = f.readline()
+
+ for l in f:
+ i.append(list(map(int, l.strip().split(','))))
+
+# 1
+res = 0
+un = []
+for l in i:
+ for j in range(len(l)):
+ if l[j] in o:
+ if not set(l[j+1:]) <= o[l[j]]:
+ un.append(l)
+ break
+ if l[j] in p:
+ if not set(l[:j]) <= p[l[j]]:
+ un.append(l)
+ break
+ else:
+ res += l[len(l)//2]
+
+# 2
+def cmp(a,b):
+ if a in o and b in o[a]: return -1
+ if b in o and a in o[a]: return 1
+ else: return 0
+
+
+res = 0
+for l in un:
+ ls = sorted(l, key=cmp_to_key(cmp))
+ res += ls[len(ls)//2]