题目:
生命游戏,简称为生命,是英国数学家约翰·何顿·康威在 1970 年发明的细胞自动机。
给定一个包含 m × n 个格子的面板,每一个格子都可以看成是一个细胞。每个细胞都具有一个初始状态:1 即为活细胞(live),或 0 即为死细胞(dead)。每个细胞与其八个相邻位置(水平,垂直,对角线)的细胞都遵循以下四条生存定律:
1.如果活细胞周围八个位置的活细胞数少于两个,则该位置活细胞死亡;
2.如果活细胞周围八个位置有两个或三个活细胞,则该位置活细胞仍然存活;
3.如果活细胞周围八个位置有超过三个活细胞,则该位置活细胞死亡;
4.如果死细胞周围正好有三个活细胞,则该位置死细胞复活;
根据当前状态,写一个函数来计算面板上所有细胞的下一个(一次更新后的)状态。下一个状态是通过将上述规则同时应用于当前状态下的每个细胞所形成的,其中细胞的出生和死亡是同时发生的。
解题思路:
该题的解题思路是统计基准细胞周围8个细胞的存活个数,从而通过个数来判断基准细胞的状态。我们从下图可以更直观的了解四条生存定律,总结一下,可以分成两个部分。
第一,活细胞->死细胞,前提条件是该细胞是活的,那么如果周围活细胞数少于两个 or 大于3个,就符合定律变成死细胞。
第二,死细胞->活细胞,前提条件该细胞死的,如果周围活细胞等于3个,就符合条件。
其实还有第三条,我们可以理解为上述两个条件的else情况,且由于状态不变,我们可以直接忽略。
难点在于,如何统计基准细胞周围活细胞的个数。我们发现在m*n的格子边缘点上,会存在周围细胞少于8个的情况。我们的可以通过补边(padding)来解决,或者直接忽略不统计。
第一种思路可以通过卷积的方式实现,即通过设置卷积核与padding后的格子进行卷积,卷积得到的值就是活细胞的个数。
代码如下
class Solution11:
def gameOfLife(self, board):
import numpy as np
r, c = len(board),len(board[0])
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)
print(board_exp)
kernel = np.array([[1,1,1],[1,0,1],[1,1,1]])
for i in range(1,r+1):
for j in range(1,c+1):
temp_sum = np.sum(kernel*board_exp[i-1:i+2, j-1:j+2])
if board_exp[i,j]==1:
if temp_sum<2 or temp_sum>3:
board[i-1][j-1]=0
if board_exp[i,j]==0:
if temp_sum==3:
board[i-1][j-1]=1
第二种思路,通过遍历所有周围的细胞并判断统计活细胞的个数。在统计的时候,我们要注意剔除格子边缘点,其周围细胞不存在的索引。
class Solution:
def gameOfLife(self, board: List[List[int]]) -> None:
"""
Do not return anything, modify board in-place instead.
"""
neighbors = [(1,0), (1,-1), (0,-1), (-1,-1), (-1,0), (-1,1), (0,1), (1,1)]
rows = len(board)
cols = len(board[0])
# 从原数组复制一份到 copy_board 中
copy_board = [[board[row][col] for col in range(cols)] for row in range(rows)]
# 遍历面板每一个格子里的细胞
for row in range(rows):
for col in range(cols):
# 对于每一个细胞统计其八个相邻位置里的活细胞数量
live_neighbors = 0
for neighbor in neighbors:
r = (row + neighbor[0])
c = (col + neighbor[1])
# 查看相邻的细胞是否是活细胞
if (r < rows and r >= 0) and (c < cols and c >= 0) and (copy_board[r][c] == 1):
live_neighbors += 1
# 规则 1 或规则 3
if copy_board[row][col] == 1 and (live_neighbors < 2 or live_neighbors > 3):
board[row][col] = 0
# 规则 4
if copy_board[row][col] == 0 and live_neighbors == 3:
board[row][col] = 1