25 lines
816 B
Python
25 lines
816 B
Python
|
with open('day25-input', 'r') as file:
|
||
|
|
data = [l.strip('\n') for l in file]
|
||
|
|
import numpy as np
|
||
|
|
import re
|
||
|
|
|
||
|
|
numbers = [[int(s) for s in re.findall(r'-?\d+', d)] for d in data]
|
||
|
|
arr = np.array(numbers, dtype=np.int64)
|
||
|
|
max_constellation_radius = 3
|
||
|
|
constellation_ids = np.arange((len(arr)), dtype=np.int64)
|
||
|
|
|
||
|
|
|
||
|
|
def points_in_range_of(index):
|
||
|
|
distances = np.abs(arr - arr[index]).sum(axis=1)
|
||
|
|
return np.argwhere(distances <= max_constellation_radius)[:,0] # Note this will include the point itself
|
||
|
|
|
||
|
|
|
||
|
|
for i in range(len(arr)):
|
||
|
|
points = points_in_range_of(i)
|
||
|
|
constellations_to_merge = constellation_ids[points]
|
||
|
|
constellation_id = constellation_ids[points].min()
|
||
|
|
for c in constellations_to_merge:
|
||
|
|
constellation_ids[constellation_ids == c] = constellation_id
|
||
|
|
|
||
|
|
print(len(np.unique(constellation_ids))) # Part 1
|