advent-of-code-2020/24/c.py

from operator import add, sub
from copy import deepcopy

directions = [x.replace('e','e ').replace('w','w ')[:-1].split(' ') for x in open('input').read().splitlines()]

def getHexAdjacent(vec):
    adj = {}
    di,dj,dk = (0,1,1)
    adj.update({
        'sw': (vec[0]+di,vec[1]+dj,vec[2]+dk),
        'ne': (vec[0]-di,vec[1]-dj,vec[2]-dk)
    })
    di,dj,dk = (1,-1,0)
    adj.update({
        'e': (vec[0]+di,vec[1]+dj,vec[2]+dk),
        'w': (vec[0]-di,vec[1]-dj,vec[2]-dk)
    })
    di,dj,dk = (1,0,1)
    adj.update({
        'se': (vec[0]+di,vec[1]+dj,vec[2]+dk),
        'nw': (vec[0]-di,vec[1]-dj,vec[2]-dk)
    })
    return adj

def toggle(boolean):
    return (False if boolean else True)

grid = {(0,0,0):{'state':False}}

vectors = []
for direction in directions:
    # print(direction)
    vec = (0,0,0)
    for dir_ in direction:
        vec = getHexAdjacent(vec)[dir_]
    grid.update({vec:{'state':toggle(grid[vec]['state'] if vec in grid else False)}})
    vectors.append(vec)

#Part 1
print(len([ x for x in grid if (grid[x]['state'] == True)])) # Black => True, white => false

#Part 2
grid_ = deepcopy(grid)  #expand our space by one hextile in each direction
for cell in grid:
    adj = getHexAdjacent(cell)
    for cell_ in adj:
        cell_ = adj[cell_]
        if cell_ not in grid:
            grid_.update({cell_:{'state':False}})

i = 0
while i < 100:
    grid = grid_
    grid_ = deepcopy(grid)
    for cell in grid:
        adj = getHexAdjacent(cell)
        c = 0
        for cell_ in adj:
            cell_ = adj[cell_]
            if cell_ in grid:
                if grid[cell_]['state'] == True:
                    c+=1
            else:
                grid_.update({cell_:{'state':False}})

        if grid[cell]['state'] == True and (c == 0 or c > 2):
            grid_[cell]['state'] = False
        elif grid[cell]['state'] == False and c == 2:
            grid_[cell]['state'] = True
    i+=1
print(len([ x for x in grid_ if (grid_[x]['state'] == True)]))