diff options
author | Tomasz Kramkowski <tk@the-tk.com> | 2021-11-24 22:25:42 +0000 |
---|---|---|
committer | Tomasz Kramkowski <tk@the-tk.com> | 2021-11-24 22:25:42 +0000 |
commit | a7a6b86002b595bc167af72606b14c67ed1bdf8f (patch) | |
tree | bff94329cf969bd9df68d3b9782fee2107db56c2 /14/solution.py | |
download | aoc2015-a7a6b86002b595bc167af72606b14c67ed1bdf8f.tar.gz aoc2015-a7a6b86002b595bc167af72606b14c67ed1bdf8f.tar.xz aoc2015-a7a6b86002b595bc167af72606b14c67ed1bdf8f.zip |
init commit
Diffstat (limited to '14/solution.py')
-rw-r--r-- | 14/solution.py | 43 |
1 files changed, 43 insertions, 0 deletions
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)) |