aoc

Advent of Code
git clone https://git.pngu.org/aoc
Log | Files | Refs | README

3_2020.py (413B)

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
import fileinput

slopes = [(1, 1), (3, 1), (5, 1), (7, 1), (1, 2)]

G = []
for line in fileinput.input():
    G.append(list(line.strip()))

ans = 1
for dc, dr in slopes:
    r = 0
    c = 0
    score = 0
    while r < len(G):
        c += dc
        r += dr
        if r < len(G) and G[r][c % len(G[r])] == "#":
            score += 1
    ans *= score
    if dc == 3 and dr == 1:
        print(score)
print(ans)