13 chain from iterable

1 files changed, 3 insertions, 5 deletions

diff --git a/13.py b/13.py
index 5c1fa34..9282fdf 100644
--- a/13.py
+++ b/13.py
@@ -1,13 +1,14 @@
 from utils import open_day
 from ast import literal_eval
 from functools import cmp_to_key
+from itertools import chain
 
 with open_day(13) as f:
     pairs = [tuple(literal_eval(v) for v in p.split('\n')) for p in f.read().rstrip().split('\n\n')]
 
 def compare(l, r):
     match (l, r):
-        case ([*l], [*r]):
+        case (  [*l],   [*r]):
             for a, b in zip(l, r):
                 cmp = compare(a,  b)
                 if cmp != 0: return cmp
@@ -22,9 +23,6 @@ for i, (l, r) in enumerate(pairs):
     if cmp < 0: p1 += i + 1
 print(p1)
 
-p2 = [[[2]],[[6]]]
-for l, r in pairs:
-    p2.append(l)
-    p2.append(r)
+p2 = [[[2]], [[6]], *chain.from_iterable(pairs)]
 p2.sort(key=cmp_to_key(compare))
 print((p2.index([[2]]) + 1) * (p2.index([[6]]) + 1))