题目描述
根据 百度百科 ,生命游戏,简称为生命,是英国数学家约翰·何顿·康威在 1970 年发明的细胞自动机。
给定一个包含 m × n 个格子的面板,每一个格子都可以看成是一个细胞。每个细胞都具有一个初始状态:1 即为活细胞(live),或 0 即为死细胞(dead)。每个细胞与其八个相邻位置(水平,垂直,对角线)的细胞都遵循以下四条生存定律:
如果活细胞周围八个位置的活细胞数少于两个,则该位置活细胞死亡;
如果活细胞周围八个位置有两个或三个活细胞,则该位置活细胞仍然存活;
如果活细胞周围八个位置有超过三个活细胞,则该位置活细胞死亡;
如果死细胞周围正好有三个活细胞,则该位置死细胞复活;
根据当前状态,写一个函数来计算面板上所有细胞的下一个(一次更新后的)状态。下一个状态是通过将上述规则同时应用于当前状态下的每个细胞所形成的,其中细胞的出生和死亡是同时发生的。
解题思路
卷积新思路
代码
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33
| import numpy as np class Solution: def gameOfLife(self, board: List[List[int]]) -> None: """ Do not return anything, modify board in-place instead. """ if not board: return c = len(board[0]) r = len(board) res = [] board_exp=np.array([[0 for _ in range(c+2)] for _ in range(r+2)]) board_exp[1:1+r,1:1+c]=np.array(board) # 取出从1开始到1+r的行数,取出从1开始到1+c的列数 kernel = np.array([[1,1,1],[1,0,1],[1,1,1]]) for i in range(r): tmp = [] for j in range(c): sum_res = np.sum(kernel * board_exp[i:i+3,j:j+3]) if board[i][j]==0: if sum_res==3: tmp.append(1) else: tmp.append(0) if board[i][j]==1: if sum_res < 2 or sum_res >3: tmp.append(0) if sum_res==2 or sum_res==3: tmp.append(1) res.append(tmp) for i in range(r): for j in range(c): board[i][j] = res[i][j]
|