Marijn/Advent-of-Code-2021
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Style 5

Browse Source
This commit is contained in:
Marijn Doeve
2021-12-05 16:40:36 +01:00
parent 8b1cbb3531
commit 31d5e20f1c
1 changed files with 8 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
5.py
View File

@@ -2,9 +2,12 @@ import numpy as np
def get_points(coord, part_one=False): def get_points(coord, part_one=False):
if part_one: if (
if abs(coord[0, 0] - coord[1, 0]) > 0 and abs(coord[0, 1] - coord[1, 1]) > 0: part_one
return [] and abs(coord[0, 0] - coord[1, 0]) > 0
and abs(coord[0, 1] - coord[1, 1]) > 0
):
return []
size = max([abs(coord[0, 0] - coord[1, 0]), abs(coord[0, 1] - coord[1, 1])]) + 1 size = max([abs(coord[0, 0] - coord[1, 0]), abs(coord[0, 1] - coord[1, 1])]) + 1
@@ -19,9 +22,10 @@ def fill_grid(coords: np.array, max_x: int, max_y: int, part_one=False):
for coord in coords: for coord in coords:
points = get_points(coord, part_one=part_one) points = get_points(coord, part_one=part_one)
for point in points: for point in points:
grid[tuple(point)] += 1 grid[tuple(point)] += 1
# print(grid.T)
return len(grid[grid >= 2]) return len(grid[grid >= 2])