birdulon
/
AdventOfCode
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

vectorize face checks

This commit is contained in:
Luke Hubmayer-Werner 2022-12-19 01:03:21 +10:30
parent d7125e971a
commit 350380f4b4
1 changed files with 6 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
2022/day18.py
View File

@ -46,20 +46,12 @@ def count_faces_2(coords: list[tuple[int,int,int]]):
field[x+1,y+1,z+1] = 1
field[flood(field, (0,0,0), connectivity=1)] = -1
surface_area = 0
for (x,y,z) in np.argwhere(field == 1):
if field[x+1,y,z] == -1:
surface_area += 1
if field[x-1,y,z] == -1:
surface_area += 1
if field[x,y+1,z] == -1:
surface_area += 1
if field[x,y-1,z] == -1:
surface_area += 1
if field[x,y,z+1] == -1:
surface_area += 1
if field[x,y,z-1] == -1:
surface_area += 1
surface_area = np.logical_and(field[:-1,:,:] == 1, field[1:,:,:] == -1).sum()
surface_area += np.logical_and(field[1:,:,:] == 1, field[:-1,:,:] == -1).sum()
surface_area += np.logical_and(field[:,:-1,:] == 1, field[:,1:,:] == -1).sum()
surface_area += np.logical_and(field[:,1:,:] == 1, field[:,:-1,:] == -1).sum()
surface_area += np.logical_and(field[:,:,:-1] == 1, field[:,:,1:] == -1).sum()
surface_area += np.logical_and(field[:,:,1:] == 1, field[:,:,:-1] == -1).sum()
return surface_area