LeetCode 289. 生命游戏
题目描述:
给定一个包含 m × n 个格子的面板,每一个格子都可以看成是一个细胞。每个细胞都具有一个初始状态:1 即为活细胞(live),或 0 即为死细胞(dead)。每个细胞与其八个相邻位置(水平,垂直,对角线)的细胞都遵循以下四条生存定律:
如果活细胞周围八个位置的活细胞数少于两个,则该位置活细胞死亡;
如果活细胞周围八个位置有两个或三个活细胞,则该位置活细胞仍然存活;
如果活细胞周围八个位置有超过三个活细胞,则该位置活细胞死亡;
如果死细胞周围正好有三个活细胞,则该位置死细胞复活;
根据当前状态,写一个函数来计算面板上所有细胞的下一个(一次更新后的)状态。下一个状态是通过将上述规则同时应用于当前状态下的每个细胞所形成的,其中细胞的出生和死亡是同时发生的。
在int数据的第二个bit上操作,可以节省空间。
class Solution { public: void gameOfLife(vector<vector<int>>& board) { int rows = board.size(); int cols = board[0].size(); int dx[] = {-1, 0, 1, -1, 1, -1, 0, 1}; int dy[] = { -1, -1, -1, 0, 0, 1, 1, 1};//方便遍历周围八个细胞 for (int i = 0; i < rows; i++) { for (int j = 0; j < cols; j++) { int sum = 0;//用于计算细胞周围有多少活细胞 for (int k = 0; k < 8; k++) { if (dx[k] + i >= 0 && dx[k] + i < rows && dy[k] + j >= 0 && dy[k] + j < cols) sum += board[dx[k] + i][dy[k] + j]&1;//若该处开始为活细胞,则++ } if (board[i][j] == 1) { if (sum == 2 || sum == 3) board[i][j] |= 2;//若活细胞继续存活,将其第二个bit改为1 } else { if (sum == 3) board[i][j] |= 2;//若死细胞复活,将其第二个bit改为1 } } } for (int i = 0; i < rows; i++) { for (int j = 0; j < cols; j++) { board[i][j] >>= 1;//将数据右移动一个bit位,用新数据覆盖旧数据 } } } };