其他分享
首页 > 其他分享> > LeetCode289 生命游戏(模拟)

LeetCode289 生命游戏(模拟)

作者:互联网

LeetCode289 生命游戏

使用扩展标签保留更新之前的信息

这里原来是0,现在是1,标记为2;原来是1,现在是0,标记为-1。保留本轮更新之前的值

最后在遍历更新为0或1

class Solution:
    def gameOfLife(self, board: List[List[int]]) -> None:
        """
        Do not return anything, modify board in-place instead.
        """
        direction = [(-1, -1), (-1, 0), (-1, 1), (0, -1), (0, 1), (1, -1), (1, 0), (1, 1)]
        m, n = len(board), len(board[0])

        def count_live(x, y):
            count = 0
            for to_x, to_y in direction:
                cur_x, cur_y = x + to_x, y + to_y
                if cur_x >= 0 and cur_x < m and cur_y >= 0 and cur_y < n:
                    if (board[cur_x][cur_y] == 1 or board[cur_x][cur_y] == -1): count += 1
            
            return count

        def count_dead(x, y):
            count = 0
            for to_x, to_y in direction:
                cur_x, cur_y = x + to_x, y + to_y
                if cur_x >= 0 and cur_x < m and cur_y >= 0 and cur_y < n:
                    if (board[cur_x][cur_y] == 0 or board[cur_x][cur_y] == 2): count += 1
            
            return count
        
        for i in range(m):
            for j in range(n):
                if board[i][j] == 0 and count_live(i, j) == 3: board[i][j] = 2
                elif board[i][j] == 1 and (count_live(i, j) < 2 or count_live(i, j) > 3): board[i][j] = -1
        
        for i in range(m):
            for j in range(n):
                if board[i][j] == 0 or board[i][j] == -1: board[i][j] = 0
                elif board[i][j] == 1 or board[i][j] == 2: board[i][j] = 1

标签:count,direction,cur,range,LeetCode289,live,board,模拟,游戏
来源: https://www.cnblogs.com/solvit/p/16445965.html