Day 10 (glamorized)

day10/part2.py

@@ -5,18 +5,18 @@ lines = (x.strip() for x in open("input.txt"))
 
 cnt = []
 OPEN, CLOSE, PENALTY = "([{<", ")]}>", [1, 2, 3, 4]
-MAP_CLOSE, MAP_PENALTY = dict(zip(OPEN, CLOSE)), dict(zip(OPEN, PENALTY))
 for line in lines:
-    success = True
-    stack = []
-    for char in line:
-        if char in OPEN:
-            stack.append(char)
-        elif MAP_CLOSE[stack.pop()] != char:
-            success = False
-            break
-    if success:
-        cnt.append(reduce(lambda p, c: p*5 + MAP_PENALTY[c], stack[::-1], 0))
+    try:
+        stack = []
+        for char in line:
+            try:
+                stack.append(OPEN.index(char))
+            except ValueError:
+                if CLOSE[stack.pop()] != char:
+                    raise ValueError("Char not found: " + char)
+        cnt.append(reduce(lambda p, c: p*5 + PENALTY[c], stack[::-1], 0))
+    except ValueError:
+        pass
 
 cnt.sort()
 print(cnt[len(cnt) // 2])