blob: f7ea63275ee175372692e134055368306ffa4767 (
plain) (
blame)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
|
L = [list(l.strip()) for l in open("input").readlines()]
def part1():
res = 0
for i, l in enumerate(L):
current = ""
has_adjacent = False
for j, m in enumerate(l + ["."]):
if not m.isdigit():
if has_adjacent:
res += int(current)
current = ""
has_adjacent = False
continue
current += m
# next line
if i + 1 < len(L) and any(
x != "." for x in L[i + 1][j - 1 : j + 2]
):
has_adjacent = True
# previous line
if i - 1 >= 0 and any(x != "." for x in L[i - 1][j - 1 : j + 2]):
has_adjacent = True
# left + right char
left = l[j - 1] if j - 1 >= 0 else "."
right = l[j + 1] if j + 1 < len(l) else "."
if (not left.isdigit() and left != ".") or (
not right.isdigit() and right != "."
):
has_adjacent = True
print(res)
def part2():
gears = {}
for i, l in enumerate(L):
current = ""
has_adjacent = False
for j, m in enumerate(l + ["."]):
if not m.isdigit():
if has_adjacent:
if gear not in gears:
gears[gear] = []
gears[gear].append(int(current))
current = ""
has_adjacent = False
continue
current += m
# next line
next = L[i + 1][j - 1 : j + 2] if i + 1 < len(L) else []
if any(x == "*" for x in next):
gear = (i + 1, j - 1 + next.index("*"))
has_adjacent = True
# previous line
prev = L[i - 1][j - 1 : j + 2] if i - 1 >= 0 else []
if any(x == "*" for x in prev):
gear = (i - 1, j - 1 + prev.index("*"))
has_adjacent = True
# left char
left = l[j - 1] if j - 1 >= 0 else "."
if left == "*":
gear = (i, j - 1)
has_adjacent = True
# right char
right = l[j + 1] if j + 1 < len(l) else "."
if right == "*":
gear = (i, j + 1)
has_adjacent = True
print(sum(gear[0] * gear[1] for gear in gears.values() if len(gear) == 2))
part1()
part2()
|
