From c13206f5aa6d66ea7197747dc1152febc81a7907 Mon Sep 17 00:00:00 2001 From: Bruno BELANYI Date: Thu, 3 Dec 2020 18:21:57 +0100 Subject: [PATCH] 2020: d03: ex1: add solution --- 2020/d03/ex1/ex1.py | 25 +++++++++++++++++++++++++ 1 file changed, 25 insertions(+) create mode 100755 2020/d03/ex1/ex1.py diff --git a/2020/d03/ex1/ex1.py b/2020/d03/ex1/ex1.py new file mode 100755 index 0000000..c964770 --- /dev/null +++ b/2020/d03/ex1/ex1.py @@ -0,0 +1,25 @@ +#!/usr/bin/env python + +import sys +from typing import List, Tuple + + +def solve(trees: List[str], delta: Tuple[int, int]) -> int: + x, y = 0, 0 + sum = 0 + while True: + x += delta[0] + y += delta[1] + if y >= len(trees): + break + sum += trees[y][x % len(trees[0])] == "#" + return sum + + +def main() -> None: + input = [line.strip() for line in sys.stdin.readlines()] + print(solve(input, (3, 1))) + + +if __name__ == "__main__": + main()