porting over AOC from previous years to a monorepo.

2020/days/03/day3.py

@@ -0,0 +1,38 @@
+#! /usr/bin/env python3
+
+import pathlib
+import sys
+import re
+from dataclasses import dataclass
+from pprint import pprint
+
+def parse(puzzle_input: str):
+    """Parse input"""
+    # returns a 2 dimentional array where true means there is a tree there.
+    return [[i=='#' for i in j] for j in puzzle_input.splitlines()]
+
+def part1(data):
+    """Solve part 1"""
+    return solveForSlope(data, 3, 1)
+
+def solveForSlope(data, right: int, down: int):
+    return sum(1 for index, row in enumerate(data[::down]) if row[(right*index)%len(row)])
+
+def part2(data):
+    """Solve part 2"""
+    return solveForSlope(data,1,1)*solveForSlope(data,3,1)*solveForSlope(data,5,1)*solveForSlope(data,7,1)*solveForSlope(data,1,2)
+
+def solve(puzzle_input):
+    """Solve the puzzle for the given input"""
+    data = parse(puzzle_input)
+    solution1 = part1(data)
+    solution2 = part2(data)
+
+    return solution1, solution2
+
+if __name__ == "__main__":
+    for path in sys.argv[1:]:
+        print(f"{path}:")
+        puzzle_input = pathlib.Path(path).read_text().strip()
+        solutions = solve(puzzle_input)
+        print("\n".join(str(solution) for solution in solutions))