LeetCode 289. 生命游戏

题目描述

根据百度百科，生命游戏，简称为生命，是英国数学家约翰·何顿·康威在1970年发明的细胞自动机。

给定一个包含 m × n 个格子的面板，每一个格子都可以看成是一个细胞。每个细胞具有一个初始状态 live（1）即为活细胞， 或 dead（0）即为死细胞。每个细胞与其八个相邻位置（水平，垂直，对角线）的细胞都遵循以下四条生存定律：

1. 如果活细胞周围八个位置的活细胞数少于两个，则该位置活细胞死亡；
2. 如果活细胞周围八个位置有两个或三个活细胞，则该位置活细胞仍然存活；
3. 如果活细胞周围八个位置有超过三个活细胞，则该位置活细胞死亡；
4. 如果死细胞周围正好有三个活细胞，则该位置死细胞复活；

根据当前状态，写一个函数来计算面板上细胞的下一个（一次更新后的）状态。下一个状态是通过将上述规则同时应用于当前状态下的每个细胞所形成的，其中细胞的出生和死亡是同时发生的。

示例:

输入: 
[
  [0,1,0],
  [0,0,1],
  [1,1,1],
  [0,0,0]
]
输出: 
[
  [0,0,0],
  [1,0,1],
  [0,1,1],
  [0,1,0]
]

进阶:

• 你可以使用原地算法解决本题吗？请注意，面板上所有格子需要同时被更新：你不能先更新某些格子，然后使用它们的更新后的值再更新其他格子。
• 本题中，我们使用二维数组来表示面板。原则上，面板是无限的，但当活细胞侵占了面板边界时会造成问题。你将如何解决这些问题？

问题分析

此题用状态机来实现常数空间复杂度。定义四种状态：状态0：死细胞转为死细胞；状态1：活细胞转为活细胞；状态2：活细胞转为死细胞；状态3：死细胞转为活细胞。这样我们就可以在任何时刻知道任何细胞的初始状态了。我们遍历二维数组的每一个细胞，判断当前细胞的周围8个细胞的初始状态，当某周围细胞的状态是1或2时，它的初始状态一定是活细胞，于是用变量count保存周围8个细胞中初始状态为活细胞的数量。于是我们便可以利用count和当前细胞的状态，更新当前细胞的状态。当二维数组所有细胞状态都更新完毕后，要将所有细胞的状态从状态机的四个状态恢复成0，1的两个状态，于是将它们的状态对2求余，所得的数就是它们现在应该的状态0或者1了。

代码实现

class Solution {
public:
    void gameOfLife(vector<vector<int>>& board) {
        int m = board.size();
        int n = board[0].size();
        int dx[] = {-1, 0, 1, 1, 1, 0, -1, -1};
        int dy[] = {-1, -1, -1, 0, 1, 1, 1, 0};
        for(int i = 0; i < m; i++){
            for(int j = 0; j < n; j++){
                int count = 0;
                for(int k = 0; k < 8; k++){
                    int x = i + dx[k];
                    int y = j + dy[k];
                    if(x >= 0 && x < m && y >= 0 && y < n && (board[x][y] == 1 || board[x][y] == 2))
                        count++; 
                }
                if(board[i][j] == 0 && count == 3)
                    board[i][j] = 3;
                else if(board[i][j] == 1 && (count < 2 || count > 3))
                    board[i][j] = 2;
            }
        }
        for(int i = 0; i < m; i++){
            for(int j = 0; j < n; j++){
                board[i][j] %= 2;
            }
        }
    }
};