289. 生命游戏
原始题目链接:https://leetcode.cn/problems/game-of-life/
根据 百度百科 , 生命游戏 ,简称为 生命 ,是英国数学家约翰·何顿·康威在 1970 年发明的细胞自动机。
给定一个包含 m × n 个格子的面板,每一个格子都可以看成是一个细胞。每个细胞都具有一个初始状态: 1 即为 活细胞 (live),或 0 即为 死细胞 (dead)。每个细胞与其八个相邻位置(水平,垂直,对角线)的细胞都遵循以下四条生存定律:
如果活细胞周围八个位置的活细胞数少于两个,则该位置活细胞死亡;
如果活细胞周围八个位置有两个或三个活细胞,则该位置活细胞仍然存活;
如果活细胞周围八个位置有超过三个活细胞,则该位置活细胞死亡;
如果死细胞周围正好有三个活细胞,则该位置死细胞复活;
下一个状态是通过将上述规则同时应用于当前状态下的每个细胞所形成的,其中细胞的出生和死亡是同时发生的。给你 m x n 网格面板 board 的当前状态,返回下一个状态。
提示:
m == board.length
n == board[i].length
1 <= m, n <= 25
board[i][j] 为 0 或 1
进阶:
你可以使用原地算法解决本题吗?请注意,面板上所有格子需要同时被更新:你不能先更新某些格子,然后使用它们的更新后的值再更新其他格子。
本题中,我们使用二维数组来表示面板。原则上,面板是无限的,但当活细胞侵占了面板边界时会造成问题。你将如何解决这些问题?
解题思路:
借用卷积的思想解题,定义卷积核,为了方便边界元素解题,填充0,填充的大小根据题意确定,当前元素的周围8个位置,所以行列各自增加2行2列来填充0,正好可以用3*3的卷积核大小进行卷积,卷积核中间的元素是0,因为算当前的元素的8个位置,所以是0。具体看代码及注释。
代码实现:
class Solution:
def gameOfLife(self, board: List[List[int]]) -> None:
"""
Do not return anything, modify board in-place instead.
"""
# 运用卷积的思想统计当前位置元素它的周围8个位置卷积后活细胞(1)的个数
import numpy as np
row, col = len(board), len(board[0])
# 卷积需要填充0,方便边界的元素卷积,周围填充一圈0
# 因为要计算周围8个位置,即3*3的卷积核大小,所以行和列都需要增加2,来填充0
new_board = np.array([[0 for _ in range(col + 2)] for _ in range(row + 2)])
# 没填充0的地方赋值为board中的元素
new_board[1 : 1 + row, 1 : 1 + col] = np.array(board)
# 定义卷积核,要统计当前元素的周围的1的个数,所以卷积核中间的元素为0,其余为1
kernel = np.array([[1, 1, 1], [1, 0, 1], [1, 1, 1]])
# 开始卷积,i和j从非填充0的地方开始遍历
for i in range(1, 1 + row):
for j in range(1, 1 + col):
# 统计当前元素的它的周围的8个位置的1的个数,即活细胞的个数
cur_sum = np.sum(kernel * new_board[i - 1 : i + 2, j - 1 : j + 2])
# 根据题目规则判断
if new_board[i][j] == 1:
if cur_sum < 2 or cur_sum > 3:
# new_board[i][j]对应没填充的数组的位置是 board[i-1][j-1]
board[i - 1][j - 1] = 0
else:
if cur_sum == 3:
board[i - 1][j - 1] = 1
参考文献:
https://leetcode.cn/problems/game-of-life/solution/xiong-mao-shua-ti-python3-bao-xue-bao-hui-cvzhong-/