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)