题目链接
题目描述:
根据 百度百科 ,生命游戏,简称为生命,是英国数学家约翰·何顿·康威在 1970 年发明的细胞自动机。
给定一个包含 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]
]
进阶:
- 你可以使用原地算法解决本题吗?请注意,面板上所有格子需要同时被更新:你不能先更新某些格子,然后使用它们的更新后的值再更新其他格子。
- 本题中,我们使用二维数组来表示面板。原则上,面板是无限的,但当活细胞侵占了面板边界时会造成问题。你将如何解决这些问题?
辅助数组
原地算法属于进阶思路,一开始先考虑如何解决问题,即使暴力也好。
题目要求我们同步更新数组,意味着我们不能像平时BFS那样局部更新数组,这是不妨设置一个辅助数组temp
,复制board
的状态,而后双循环判断temp
内每一个细胞四周状态对应更新board
的状态,曲线救国~
注意:这次方向不再是四个方向,而是八个,设置方向数组的时候要小心。
代码
class Solution {
public:
void gameOfLife(vector<vector<int>>& board) {
vector<vector<int>> temp(board);
int dx[8] = {0, 0, 1, -1, -1, -1, 1, 1};
int dy[8] = {1, -1, 0, 0, -1, 1, -1, 1};
int m = board.size(), n = board[0].size();
for(int i = 0; i < m; i++){
for(int j = 0; j < n; j++){
int live = 0;
for(int k = 0; k < 8; k++){
int newX = i + dx[k];
int newY = j + dy[k];
if(newX < 0 || newX >= m || newY < 0 || newY >= n)
continue;
if(temp[newX][newY] == 1)
live++;
}
if(temp[i][j] == 1){
if(live < 2 || live > 3)
board[i][j] = 0;
}
else{
if(live == 3)
board[i][j] = 1;
}
}
}
return;
}
};
原地&位运算
根据评论区大神的思路,我们还可以用位运算的方式实现原地更新。我们知道int类型存储多比特位,而根据题目可知初始状态数组元素非0即1。所以最低位存储初始状态,我们不妨利用倒数第二位存储更新后的状态,利用&
运算直接找到最后一位求解四周活细胞的数量,而后做出判断,最后向右移动一位即可获得更新后的整个数组状态。
真是太有才了,吾辈只能多加练习~
代码
class Solution {
public:
void gameOfLife(vector<vector<int>>& board) {
int dx[8] = {0, 0, 1, -1, -1, -1, 1, 1};
int dy[8] = {1, -1, 0, 0, -1, 1, -1, 1};
int m = board.size(), n = board[0].size();
for(int i = 0; i < m; i++){
for(int j = 0; j < n; j++){
int live = 0;
for(int k = 0; k < 8; k++){
int newX = i + dx[k];
int newY = j + dy[k];
if(newX < 0 || newX >= m || newY < 0 || newY >= n)
continue;
live += board[newX][newY] & 1;
}
if((board[i][j] & 1) > 0){
if(live == 2 || live == 3)
board[i][j] = 0b11;
//原数据0b01,live > 3 or live < 2都会死,符合原数据状态,无需更改
}
else{
if(live == 3)
board[i][j] = 0b10;
}
}
}
for(int i = 0; i < m; i++){
for(int j = 0; j < n; j++){
board[i][j] >>= 1;
}
}
return;
}
};
原地&自定义规则
还有一种方法就是自定义状态实现原地更新。由于初始状态下数组只有0
和1
两个状态,不妨自定义多个符合题意的状态完成算法。规则如下:
- 初始状态是活细胞,四周的活细胞总数
<2
或>3
,更新后活细胞死亡,设置状态为-1
; - 初始状态是活细胞,四周的活细胞总数
==2
或==3
,更新后活细胞依旧存活,状态不变,仍为1
; - 初始状态是死细胞,四周的活细胞总数
==3
,更新后死细胞复活,设置状态为2
; - 其余题中未特殊说明的情况,均保持原有状态不变。
代码
class Solution {
public:
void gameOfLife(vector<vector<int>>& board) {
int dx[8] = {0, 0, 1, -1, -1, -1, 1, 1};
int dy[8] = {1, -1, 0, 0, -1, 1, -1, 1};
int m = board.size(), n = board[0].size();
for(int i = 0; i < m; i++){
for(int j = 0; j < n; j++){
int live = 0;
for(int k = 0; k < 8; k++){
int newX = i + dx[k];
int newY = j + dy[k];
if(newX < 0 || newX >= m || newY < 0 || newY >= n)
continue;
if(board[newX][newY] == 1 || board[newX][newY] == -1)
live++;
}
if(board[i][j] == 1 || board[i][j] == -1){
if(live < 2 || live > 3)
board[i][j] = -1;
}
else{
if(live == 3)
board[i][j] = 2;
}
}
}
for(int i = 0; i < m; i++){
for(int j = 0; j < n; j++){
if(board[i][j] == 2)
board[i][j] = 1;
else if(board[i][j] == -1)
board[i][j] = 0;
}
}
return;
}
};
ps:原地算法没有辅助数组快,哭了。。。
参考链接
如果有错误或者不严谨的地方,请务必给予指正,十分感谢。
本人blog:http://breadhunter.gitee.io