Prison Cells After N Days

PrisonCellsAfterNDays.py

class Solution:
    def prisonAfterNDays(self, cells: List[int], N: int) -> List[int]:
        memory = {}
        for it in range(N):
            ocells = cells[:]
            for i in range(1,7):
                if ocells[i-1]==ocells[i+1]:
                    cells[i] = 1
                else:
                    cells[i] = 0
            cells[0], cells[-1] = 0, 0
            
            if memory and memory[0]==cells:
                break
            
            memory[it] = cells[:]
        rep = len(memory)
        return memory[(N-1)%rep]