1.原地算法
由于原二维矩阵中只包含0,和1,0表示死亡,1表示存活
我们可以设置如下状态:其中第一个位置表示新的状态,第二个位置表示旧的状态
0 0:新状态死亡 旧状态死亡
0 1:新状态死亡 旧状态存活
1 0:新状态存活 旧状态死亡
1 1:新状态存活,旧状态存活
对于当前位置的元素,如果要获取其新状态,则右移1位,如果要获取其旧状态,则&1
因此,对于最后的状态更新,只需要所有元素都右移一位即可。
对于上述的四种状态,只有两个情况需要特殊处理: 1 0和1 1
因为0 0与0 1更新状态时,无论其初始状态是什么,右移一位都可以直接得到其最终状态,即死亡。
1)遍历矩阵,计算当前元素的周围细胞存活情况,如果符号1 1 的状态要求,则令当前位置的元素值为3(二进制为1 1),
如果符号1 0 的状态要求,则令当前位置的元素值为2(二进制为1 0)。
2)更新矩阵
更新矩阵时,所有元素值右移一位即可
代码如下:
class Solution {
public:
void gameOfLife(vector<vector<int>>& board) {
for (int row = 0; row < board.size(); ++row) {
for (int column = 0; column < board[0].size(); ++column) {
getState(board, row, column);
}
}
// refresh
for (int row = 0; row < board.size(); ++row) {
for (int column = 0; column < board[0].size(); ++column) {
board[row][column] = board[row][column] >> 1;
}
}
}
void getState(vector<vector<int>>& board, int row, int column) {
int pos[][2] = {{-1, -1}, {-1, 0}, {-1, 1}, {0, -1}, {0, 1}, {1, -1}, {1, 0}, {1, 1}};
int count = 0;
//统计当前位置周围的情况
for (int k = 0; k < 8; ++k) {
int i = pos[k][0] + row;
int j = pos[k][1] + column;
if (i < 0 || i >= board.size() || j < 0 || j >= board[0].size()) {
continue;
}
if (board[i][j] &1) {
count++;
}
}
if (board[row][column] == 1 && (count == 2 || count == 3)) {
board[row][column] = 3;//原来为1,现在为3,二进制为1 1,
} else if (board[row][column] == 0 && count == 3) {
board[row][column] = 2;
}
}
};