Solved 2024/08 P1
This commit is contained in:
parent
f3f88f47d1
commit
88f2b09174
@ -5,37 +5,37 @@ sys.path.insert(0, '../../')
|
|||||||
from fred import list2int,get_re,nprint,lprint,grid_valid,loadFile,subTuples,addTuples,toGrid,get_value_in_direction
|
from fred import list2int,get_re,nprint,lprint,grid_valid,loadFile,subTuples,addTuples,toGrid,get_value_in_direction
|
||||||
start_time = time.time()
|
start_time = time.time()
|
||||||
|
|
||||||
input_f = 'test'
|
input_f = 'test3'
|
||||||
|
|
||||||
part = 1
|
part = 2
|
||||||
#########################################
|
#########################################
|
||||||
# #
|
# #
|
||||||
# Part 1 #
|
# Part 1 #
|
||||||
# #
|
# #
|
||||||
#########################################
|
#########################################
|
||||||
|
|
||||||
def findAntinodes(grid:list,pos:set,current_node:set):
|
|
||||||
print('Found',current_node,'at',pos)
|
|
||||||
antinodes = []
|
|
||||||
for row in range(pos[0],len(grid)):
|
|
||||||
for col in range(0,len(grid[1])):
|
|
||||||
print(row,col)
|
|
||||||
|
|
||||||
if get_value_in_direction(grid,(row,col)) == current_node and (row,col) != pos:
|
|
||||||
print('Matched with',(row,col))
|
|
||||||
|
|
||||||
s = subTuples((row,col-1),pos)
|
|
||||||
a = addTuples(s,(row,col+1))
|
|
||||||
if grid_valid(s[0],s[1],grid):
|
|
||||||
print('Antinode at',s)
|
|
||||||
antinodes.append(s)
|
|
||||||
if grid_valid(a[0],a[1],grid):
|
|
||||||
print('Antinode at',a)
|
|
||||||
antinodes.append(a)
|
|
||||||
|
|
||||||
return antinodes
|
|
||||||
|
|
||||||
if part == 1:
|
if part == 1:
|
||||||
|
def findAntinodes(grid:list,pos:set,current_node:set):
|
||||||
|
#print('Found',current_node,'at',pos)
|
||||||
|
antinodes = []
|
||||||
|
for row in range(pos[0],len(grid)):
|
||||||
|
for col in range(0,len(grid[1])):
|
||||||
|
|
||||||
|
if get_value_in_direction(grid,(row,col)) == current_node and (row,col) != pos:
|
||||||
|
#print('Matched with',(row,col))
|
||||||
|
|
||||||
|
dist = subTuples((row,col),pos)
|
||||||
|
add = addTuples(dist,(row,col))
|
||||||
|
sub = subTuples(pos,dist)
|
||||||
|
if grid_valid(sub[0],sub[1],grid):
|
||||||
|
#print('Antinode at',sub)
|
||||||
|
antinodes.append(sub)
|
||||||
|
if grid_valid(add[0],add[1],grid):
|
||||||
|
#print('Antinode at',add)
|
||||||
|
antinodes.append(add)
|
||||||
|
|
||||||
|
return antinodes
|
||||||
|
|
||||||
grid = toGrid(input_f)
|
grid = toGrid(input_f)
|
||||||
|
|
||||||
nprint(grid)
|
nprint(grid)
|
||||||
@ -47,7 +47,7 @@ if part == 1:
|
|||||||
current_node = get_value_in_direction(grid,(row,col))
|
current_node = get_value_in_direction(grid,(row,col))
|
||||||
if current_node != '.':
|
if current_node != '.':
|
||||||
antinodes += findAntinodes(grid,(row,col),current_node)
|
antinodes += findAntinodes(grid,(row,col),current_node)
|
||||||
print(antinodes)
|
#print(antinodes)
|
||||||
for idx, i in enumerate(grid):
|
for idx, i in enumerate(grid):
|
||||||
for jdx, j in enumerate(i):
|
for jdx, j in enumerate(i):
|
||||||
if (idx, jdx) in antinodes:
|
if (idx, jdx) in antinodes:
|
||||||
@ -56,6 +56,8 @@ if part == 1:
|
|||||||
print(grid[idx][jdx], end=' ') # Regular grid element
|
print(grid[idx][jdx], end=' ') # Regular grid element
|
||||||
print()
|
print()
|
||||||
|
|
||||||
|
print(len(set(antinodes)))
|
||||||
|
|
||||||
|
|
||||||
#########################################
|
#########################################
|
||||||
# #
|
# #
|
||||||
@ -63,6 +65,47 @@ if part == 1:
|
|||||||
# #
|
# #
|
||||||
#########################################
|
#########################################
|
||||||
if part == 2:
|
if part == 2:
|
||||||
exit()
|
def findAntinodes(grid:list,pos:set,current_node:set):
|
||||||
|
#print('Found',current_node,'at',pos)
|
||||||
|
antinodes = []
|
||||||
|
for row in range(pos[0],len(grid)):
|
||||||
|
for col in range(0,len(grid[1])):
|
||||||
|
|
||||||
print("--- %s seconds ---" % (time.time() - start_time))
|
if get_value_in_direction(grid,(row,col)) == current_node and (row,col) != pos:
|
||||||
|
#print('Matched with',(row,col))
|
||||||
|
|
||||||
|
dist = subTuples((row,col),pos)
|
||||||
|
add = addTuples(dist,(row,col))
|
||||||
|
sub = subTuples(pos,dist)
|
||||||
|
if grid_valid(sub[0],sub[1],grid):
|
||||||
|
#print('Antinode at',sub)
|
||||||
|
antinodes.append(sub)
|
||||||
|
if grid_valid(add[0],add[1],grid):
|
||||||
|
#print('Antinode at',add)
|
||||||
|
antinodes.append(add)
|
||||||
|
|
||||||
|
return antinodes
|
||||||
|
|
||||||
|
grid = toGrid(input_f)
|
||||||
|
|
||||||
|
nprint(grid)
|
||||||
|
|
||||||
|
size = (len(grid),len(grid[0]))
|
||||||
|
antinodes = []
|
||||||
|
for row in range(0,size[0]):
|
||||||
|
for col in range(0,size[1]):
|
||||||
|
current_node = get_value_in_direction(grid,(row,col))
|
||||||
|
if current_node != '.':
|
||||||
|
antinodes += findAntinodes(grid,(row,col),current_node)
|
||||||
|
#print(antinodes)
|
||||||
|
for idx, i in enumerate(grid):
|
||||||
|
for jdx, j in enumerate(i):
|
||||||
|
if (idx, jdx) in antinodes:
|
||||||
|
print('#', end=' ') # Print sign
|
||||||
|
else:
|
||||||
|
print(grid[idx][jdx], end=' ') # Regular grid element
|
||||||
|
print()
|
||||||
|
|
||||||
|
print(len(set(antinodes)))
|
||||||
|
|
||||||
|
print("--- %s seconds ---" % (time.time() - start_time))
|
||||||
|
Loading…
Reference in New Issue
Block a user