From ea8d287db583f2bc7e5f8d4e4ed178a1b47465e1 Mon Sep 17 00:00:00 2001 From: Sijmen Date: Wed, 20 Dec 2023 16:00:10 +0100 Subject: [PATCH] day 6 hype --- day06.py | 22 ++++++++++++++++++++++ 1 file changed, 22 insertions(+) create mode 100644 day06.py diff --git a/day06.py b/day06.py new file mode 100644 index 0000000..c23f3b5 --- /dev/null +++ b/day06.py @@ -0,0 +1,22 @@ +import fileinput +from math import ceil, floor, sqrt + +def race(time, dist): + x1 = (time - sqrt(time * time - 4 * dist)) / 2 + x2 = (time + sqrt(time * time - 4 * dist)) / 2 + x1, x2 = floor(x1) + 1, ceil(x2) - 1 + return x2 - x1 + 1 + +input = list(fileinput.input()) +times = [int(x) for x in input[0][11:].split()] +distances = [int(x) for x in input[1][11:].split()] + +part1 = 1 +for time, dist in zip(times, distances): + part1 *= race(time, dist) + +time = int(input[0][11:].replace(" ", "")) +dist = int(input[1][11:].replace(" ", "")) +part2 = race(time, dist) + +print(part1, part2)