matthias/aoc2020
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

use 2d array

Browse Source
This commit is contained in:
Matthias Bilger
2020-12-11 08:15:36 +01:00
parent c882ea9278
commit 3a96aff0a3
1 changed files with 30 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
day11/module.py
View File

@@ -12,12 +12,12 @@ def pw(line):
def p1():
inp = get_input(pw)
grid = {}
width = len(inp[0])
height = len(inp)
grid = [['' for y in x] for x in inp]
for y, sample in enumerate(inp):
for x, o in enumerate(sample):
grid[(x,y)] = o
grid[y][x] = o
changed = True
adj = [(-1, 0), (0, -1), (1, 0), (0, 1), (1,-1), (1,1), (-1, 1), (-1, -1)]
@@ -30,28 +30,29 @@ def p1():
count = 0
for ax, ay in adj:
try:
if grid[(x+ax, y+ay)] == '#':
if grid[y+ay][x+ax] == '#':
count += 1
except:
pass
if count == 0 and grid[(x,y)] == 'L':
newgrid[(x,y)] = '#'
if count == 0 and grid[y][x] == 'L':
newgrid[y][x] = '#'
changed = True
ccnt += 1
if count >= 4 and grid[(x,y)] == '#':
newgrid[(x,y)] = 'L'
if count >= 4 and grid[y][x] == '#':
newgrid[y][x] = 'L'
changed = True
ccnt += 1
grid = newgrid
count = 0
if draw():
for y in range(height):
for x in range(width):
print(grid[(x,y)], end='')
if grid[(x,y)] == '#':
if draw():
print(grid[y][x], end='')
if grid[y][x] == '#':
count += 1
if draw():
print()
print(count)
@@ -59,12 +60,12 @@ def p1():
def p2(inp):
grid = {}
width = len(inp[0])
height = len(inp)
grid = [['' for y in x] for x in inp]
for y, sample in enumerate(inp):
for x, o in enumerate(sample):
grid[(x,y)] = o
grid[y][x] = o
changed = True
adj = [(-1, 0), (0, -1), (1, 0), (0, 1), (1,-1), (1,1), (-1, 1), (-1, -1)]
@@ -74,7 +75,7 @@ def p2(inp):
changed = False
for y in range(height):
for x in range(width):
if grid[(x,y)] == '.':
if grid[y][x] == '.':
continue
count = 0
for ax, ay in adj:
@@ -82,7 +83,7 @@ def p2(inp):
ay1 = ay
while True:
try:
if grid[(x+ax, y+ay)] == '.':
if grid[y+ay][x+ax] == '.':
ax += ax1
ay += ay1
else:
@@ -90,28 +91,29 @@ def p2(inp):
except:
break
try:
if grid[(x+ax, y+ay)] == '#':
if grid[y+ay][x+ax] == '#':
count += 1
except:
pass
if count == 0 and grid[(x,y)] == 'L':
newgrid[(x,y)] = '#'
if count == 0 and grid[y][x] == 'L':
newgrid[y][x] = '#'
changed = True
ccnt += 1
if count >= 5 and grid[(x,y)] == '#':
newgrid[(x,y)] = 'L'
if count >= 5 and grid[y][x] == '#':
newgrid[y][x] = 'L'
changed = True
ccnt += 1
grid = newgrid
count = 0
if draw():
for y in range(height):
for x in range(width):
print(grid[(x,y)], end='')
if grid[(x,y)] == '#':
if draw():
print(grid[y][x], end='')
if grid[y][x] == '#':
count += 1
if draw():
print()
print(count)