I took a help from here so huge credit goes to the author. My repository contains a notebook where I have updated previous day's challenges also.
My solution is below lets share yours.
Challenge 1
with open("day11.txt", "r") as fp:
lines = [line.rstrip() for line in fp.readlines()]
lines = [list(line) for line in lines]
rows, cols = len(lines), len(lines[0])
deltas = [(-1,-1),(-1,0),(-1,1),(0,-1),(0,1),(1,-1),(1,0),(1,1)]
def count_occupied(r, c, grid):
count=0
for i,j in deltas:
xi,xj=r+i,c+j
if 0<=xi<rows and 0<=xj<cols and grid[xi][xj]=='#':
count+=1
return count
def check_occupied(lines, thresh = 4):
while True:
valid = True
temp_grid=[r.copy() for r in lines]
for i, r in enumerate(temp_grid):
for j, c in enumerate(r):
count = count_occupied(i, j, temp_grid)
if c=='L' and count==0:
lines[i][j]='#'
elif c=='#' and count>=thresh:
lines[i][j]='L'
valid&=(r[j]==lines[i][j])
if valid:
break
ans=0
for i in range(rows):
for j in range(cols):
if lines[i][j]=='#':
ans+=1
print(f"There are {ans} valid seats.")
check_occupied(lines)
Challenge 2
with open("day11.txt", "r") as fp:
lines = [line.rstrip() for line in fp.readlines()]
lines = [list(line) for line in lines]
def count_occupied2(r, c, grid):
count=0
for i,j in deltas:
xi,xj=r+i,c+j
while 0<=xi<rows and 0<=xj<cols:
if grid[xi][xj]=='#':
count+=1
break
elif grid[xi][xj] == 'L':
break
xi+=i
xj+=j
return count
def check_occupied2(lines, thresh = 5):
while True:
valid = True
temp_grid=[r.copy() for r in lines]
for i, r in enumerate(temp_grid):
for j, c in enumerate(r):
count = count_occupied2(i, j, temp_grid)
if c=='L' and count==0:
lines[i][j]='#'
elif c=='#' and count>=thresh:
lines[i][j]='L'
valid&=(r[j]==lines[i][j])
if valid:
break
ans=0
for i in range(rows):
for j in range(cols):
if lines[i][j]=='#':
ans+=1
print(f"There are {ans} valid seats.")
check_occupied2(lines)
Top comments (0)