-
-
Notifications
You must be signed in to change notification settings - Fork 0
/
day_11.py
76 lines (61 loc) · 1.98 KB
/
day_11.py
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
import itertools
from collections import Counter, defaultdict, deque
import aoc_helper
from aoc_helper import (
Grid,
PrioQueue,
SparseGrid,
decode_text,
extract_ints,
extract_iranges,
extract_ranges,
extract_uints,
frange,
irange,
iter,
list,
map,
multirange,
range,
search,
tail_call,
)
raw = aoc_helper.fetch(11, 2023)
def parse_raw(raw: str):
return Grid.from_string(raw)
data = parse_raw(raw)
# providing this default is somewhat of a hack - there isn't any other way to
# force type inference to happen, AFAIK - but this won't work with standard
# collections (list, set, dict, tuple)
def part_one(data=data, scale=2):
dist = 0
points_captured = 0
distance_travelled = 0
for row in data.data:
points_in_row = row.sum()
dist += distance_travelled * points_in_row
if points_in_row == 0:
distance_travelled += scale * points_captured
else:
points_captured += points_in_row
distance_travelled += points_captured
points_captured = 0
distance_travelled = 0
for row in data.transpose().data:
points_in_col = row.sum()
dist += distance_travelled * points_in_col
if points_in_col == 0:
distance_travelled += scale * points_captured
else:
points_captured += points_in_col
distance_travelled += points_captured
return dist
aoc_helper.lazy_test(day=11, year=2023, parse=parse_raw, solution=part_one)
# providing this default is somewhat of a hack - there isn't any other way to
# force type inference to happen, AFAIK - but this won't work with standard
# collections (list, set, dict, tuple)
def part_two(data=data):
return part_one(data, 1_000_000)
# aoc_helper.lazy_test(day=11, year=2023, parse=parse_raw, solution=part_two)
aoc_helper.lazy_submit(day=11, year=2023, solution=part_one, data=data)
aoc_helper.lazy_submit(day=11, year=2023, solution=part_two, data=data)