Day 06

day06/part1.py

@@ -0,0 +1,21 @@
+#!/usr/bin/env python3
+import math
+import re
+
+lines = (x.strip() for x in open("input.txt"))
+times = []
+records = []
+
+for line in lines:
+    if line.startswith("Time"):
+        times = tuple(map(int, re.findall("\d+", line)))
+    else:
+        records = tuple(map(int, re.findall("\d+", line)))
+
+result = 1
+for t, r in zip(times, records):
+    pmin = (t-(t**2-4*r)**0.5)/2
+    pmax = (t+(t**2-4*r)**0.5)/2
+    result *= math.floor(pmax) - math.ceil(pmin) + 1
+
+print(result)