题目
给定一个包含 m × n 个格子的面板,每一个格子都可以看成是一个细胞。每个细胞都具有一个初始状态:1 即为活细胞(live),或 0 即为死细胞(dead)。每个细胞与其八个相邻位置(水平,垂直,对角线)的细胞都遵循以下四条生存定律:
- 如果活细胞周围八个位置的活细胞数少于两个,则该位置活细胞死亡;
- 如果活细胞周围八个位置有两个或三个活细胞,则该位置活细胞仍然存活;
- 如果活细胞周围八个位置有超过三个活细胞,则该位置活细胞死亡;
- 如果死细胞周围正好有三个活细胞,则该位置死细胞复活;
根据当前状态,写一个函数来计算面板上所有细胞的下一个(一次更新后的)状态。下一个状态是通过将上述规则同时应用于当前状态下的每个细胞所形成的,其中细胞的出生和死亡是同时发生的。
示例:
输入:
[
[0,1,0],
[0,0,1],
[1,1,1],
[0,0,0]
]
输出:
[
[0,0,0],
[1,0,1],
[0,1,1],
[0,1,0]
]
进阶:
- 你可以使用原地算法解决本题吗?请注意,面板上所有格子需要同时被更新:你不能先更新某些格子,然后使用它们的更新后的值再更新其他格子。
- 本题中,我们使用二维数组来表示面板。原则上,面板是无限的,但当活细胞侵占了面板边界时会造成问题。你将如何解决这些问题?
解题思路
解法一:创建新数组
先将原数组A复制到一个新的数组B中,然后按照题意规则,遍历数组B,并在A中进行修改。
复杂度分析:
时间复杂度:O(mn),其中 m 和 n 分别为 board 的行数和列数。
空间复杂度:O(mn),为复制数组占用的空间。
解法二:原地数组
采取一些额外的规则,如果细胞周围有一个活细胞,那么就加10。比如31,表示此细胞是活的,并且周围有3个活细胞;又比如20,表示此细胞是死的,周围有两个活细胞。
复杂度分析:
时间复杂度:O(mn),其中 m,n 分别为 board 的行数和列数。
空间复杂度:O(1),除原数组外只需要常数的空间存放若干变量。
代码
解法一:创建新数组
class Solution:
# 用来判断细胞是活的还是死的
def live_death(self, board, i, j):
m = len(board)
n = len(board[0])
directions = [(-1,0), (1,0), (0,-1), (0,1), (-1,-1), (1,-1), (-1,1), (1,1)] # 上、下、左、右、左上、左下、右上、右下
cnt = 0
for dx, dy in directions:
if 0<=i+dx<m and 0<=j+dy<n and board[i+dx][j+dy] == 1:
cnt += 1
if board[i][j] == 1 and cnt<2:
return False
elif board[i][j] == 1 and 2<=cnt<=3:
return True
elif board[i][j] == 1 and cnt>3:
return False
elif board[i][j] == 0 and cnt == 3:
return True
else:
return False
def gameOfLife(self, board: List[List[int]]) -> None:
"""
Do not return anything, modify board in-place instead.
"""
m = len(board)
n = len(board[0])
board_bak = [[board[a][b] for b in range(n)] for a in range(m)]
for i in range(m):
for j in range(n):
if self.live_death(board_bak, i, j):
board[i][j] = 1
else:
board[i][j] = 0
解法二:原地数组
class Solution:
# 用来改变周围细胞
def affect(self, board, i, j):
m = len(board)
n = len(board[0])
directions = [(-1,0), (1,0), (0,-1), (0,1), (-1,-1), (1,-1), (-1,1), (1,1)] # 上、下、左、右、左上、左下、右上、右下
for dx, dy in directions:
if 0<=i+dx<m and 0<=j+dy<n and board[i+dx][j+dy]%10 == 1:
board[i][j] += 10
else:
continue
def gameOfLife(self, board: List[List[int]]) -> None:
"""
Do not return anything, modify board in-place instead.
"""
m = len(board)
n = len(board[0])
for i in range(m):
for j in range(n):
self.affect(board, i, j)
for i in range(m):
for j in range(n):
value = board[i][j]
if value//10 < 2 and value%10 == 1:
board[i][j] = 0
elif 2<= value//10 <= 3 and value%10 == 1:
board[i][j] = 1
elif value//10>3 and value%10 == 1:
board[i][j] = 0
elif value//10==3 and value%10 == 0:
board[i][j] = 1
else:
board[i][j] = 0