-
-
Notifications
You must be signed in to change notification settings - Fork 0
/
day_25.py
66 lines (52 loc) · 1.59 KB
/
day_25.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
from collections import Counter, defaultdict, deque
from math import prod
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,
)
from networkx import Graph, connected_components, minimum_edge_cut
raw = aoc_helper.fetch(25, 2023)
def parse_raw(raw: str):
connections = defaultdict(set)
for k, vs in (
list(raw.splitlines())
.mapped(lambda s: s.split(": "))
.starmapped(lambda k, vs: (k, set(vs.split())))
):
connections[k] |= vs
for v in vs:
connections[v] |= {k}
return Graph(connections)
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):
edges = minimum_edge_cut(data)
data.remove_edges_from(edges)
a, b = connected_components(data)
return len(a) * len(b)
# aoc_helper.lazy_test(day=25, 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):
...
aoc_helper.lazy_submit(day=25, year=2023, solution=part_one, data=data)
aoc_helper.lazy_submit(day=25, year=2023, solution=part_two, data=data)