You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
93 lines
2.0 KiB
93 lines
2.0 KiB
"""
|
|
Advent of Code 2021 - Day 02
|
|
|
|
Run with:
|
|
python puzzles/day02.py inputs/day02.txt
|
|
|
|
This is overly complicated to keep tests similar while also using dataclasses for learning
|
|
"""
|
|
|
|
import pathlib
|
|
import sys
|
|
from typing import List, Tuple
|
|
|
|
from dataclasses import dataclass
|
|
|
|
|
|
@dataclass
|
|
class Location:
|
|
horizontal = 0
|
|
depth = 0
|
|
aim = 0
|
|
|
|
def run(self, inputs: List[str]) -> int:
|
|
for line in inputs:
|
|
# forward 20 -> self.forward(20)
|
|
step = line.split()
|
|
getattr(self, step[0])(int(step[1]))
|
|
|
|
return self.calculate()
|
|
|
|
def calculate(self) -> int:
|
|
return self.horizontal * self.depth
|
|
|
|
|
|
class Part1(Location):
|
|
def forward(self, distance: int) -> None:
|
|
self.horizontal += distance
|
|
|
|
def up(self, distance: int) -> None:
|
|
self.depth -= distance
|
|
|
|
def down(self, distance: int) -> None:
|
|
self.depth += distance
|
|
|
|
|
|
class Part2(Location):
|
|
def forward(self, distance: int) -> None:
|
|
self.horizontal += distance
|
|
self.depth += self.aim * distance
|
|
|
|
def up(self, distance: int) -> None:
|
|
self.aim -= distance
|
|
|
|
def down(self, distance: int) -> None:
|
|
self.aim += distance
|
|
|
|
|
|
def part1(inputs: List[int]) -> int:
|
|
puzzle = Part1()
|
|
return puzzle.run(inputs)
|
|
|
|
|
|
def part2(inputs: List[int]) -> int:
|
|
puzzle = Part2()
|
|
return puzzle.run(inputs)
|
|
|
|
|
|
def parse(inputs: str) -> List[str]:
|
|
"""Parse the input string"""
|
|
return inputs.split("\n")
|
|
|
|
|
|
def solve(path: str) -> Tuple[int, int]:
|
|
"""Solve the puzzle"""
|
|
puzzle_input = parse(pathlib.Path(path).read_text().strip())
|
|
part1_result = part1(puzzle_input)
|
|
part2_result = part2(puzzle_input)
|
|
|
|
return part1_result, part2_result
|
|
|
|
|
|
def main() -> None:
|
|
for path in sys.argv[1:]:
|
|
print(f"Input File: {path}")
|
|
|
|
part1_result, part2_result = solve(path)
|
|
|
|
print(f"Part 1 Result: {part1_result}")
|
|
print(f"Part 2 Result: {part2_result}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|