From a7a6b86002b595bc167af72606b14c67ed1bdf8f Mon Sep 17 00:00:00 2001 From: Tomasz Kramkowski Date: Wed, 24 Nov 2021 22:25:42 +0000 Subject: init commit --- 14/solution.py | 43 +++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 43 insertions(+) create mode 100644 14/solution.py (limited to '14/solution.py') diff --git a/14/solution.py b/14/solution.py new file mode 100644 index 0000000..9ce32dd --- /dev/null +++ b/14/solution.py @@ -0,0 +1,43 @@ +from dataclasses import dataclass + +@dataclass +class Reindeer: + name: str + speed: int + fly_time: int + rest_time: int + def distance_after(self, seconds): + cycle_time = self.fly_time + self.rest_time + cycles = seconds // cycle_time + remaining_time = seconds % cycle_time + if remaining_time > self.fly_time: + remaining_time = self.fly_time + return cycles * self.speed * self.fly_time + remaining_time * self.speed + +def part1(reindeer: list[Reindeer]) -> int: + return max(r.distance_after(2503) for r in reindeer) + +def part2(reindeer: list[Reindeer]) -> int: + points = [0] * len(reindeer) + for s in range(1, 2504): + distances = [r.distance_after(s) for r in reindeer] + max_dist = max(distances) + for i, d in enumerate(distances): + if d == max_dist: + points[i] += 1 + return max(points) + +if __name__ == '__main__': + reindeer = [ + Reindeer('Vixen', 8, 8, 53), + Reindeer('Blitzen', 13, 4, 49), + Reindeer('Rudolph', 20, 7, 132), + Reindeer('Cupid', 12, 4, 43), + Reindeer('Donner', 9, 5, 38), + Reindeer('Dasher', 10, 4, 37), + Reindeer('Comet', 3, 37, 76), + Reindeer('Prancer', 9, 12, 97), + Reindeer('Dancer', 37, 1, 36), + ] + print(part1(reindeer)) + print(part2(reindeer)) -- cgit v1.2.3-70-g09d2