問題内容

有名なライフゲームの挙動を実行するプログラムを記述するもの

解答例

sample

class Solution:
    def gameOfLife(self, board: List[List[int]]) -> None:
        

        directions = [(1,0), (1,-1), (0,-1), (-1,-1), (-1,0), (-1,1), (0,1), (1,1)]
        
        for i in range(len(board)):
            for j in range(len(board[0])):
                live = 0                
                for x, y in directions: 
                    if ( i + x < len(board) and i + x >= 0 ) and ( j + y < len(board[0]) and j + y >=0 ) and abs(board[i + x][j + y]) == 1:
                        live += 1
                if board[i][j] == 1 and (live < 2 or live > 3):    
                    board[i][j] = -1
                if board[i][j] == 0 and live == 3:                  
                    board[i][j] = 2
        for i in range(len(board)):
            for j in range(len(board[0])):
                board[i][j] = 1 if(board[i][j] > 0) else 0
        return board

この問題のポイント

この問題で難しいのは、あるマスについて次の状態を判定し上書きする際、他のマスの判定に影響を与えないようにするところ。その点で解答例内の注目すべきところは以下の部分。

sample

abs(board[i + x][j + y]) == 1:

これによって絶対値が１になればカウントすることになる。これによって、判定の際にカウントされるべきものは１か‐１であれば良くなる。つまり、今回カウントされる中で次も生存するものと死亡するものでラベルを分けることができる。

同じような仕組みで

sample

if board[i][j] == 0 and live == 3:                  
    board[i][j] = 2

もある。これは、今回カウントされない中で次回は生存するものと死亡するものというラベル分けに役立っている。