LeetCode 289. 生命游戏（C++）

根据百度百科，生命游戏，简称为生命，是英国数学家约翰·何顿·康威在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]
]

C++

class Solution {
public:
    void gameOfLife(vector<vector<int>>& board) 
    {
        int m=board.size();
        if(0==m)
        {
            return;
        }
        int n=board[0].size();
        if(0==n)
        {
            return;
        }
        vector<vector<int>> flag(m,vector<int>(n,0));
        for(int i=0;i<m;i++)
        {
            for(int j=0;j<n;j++)
            {
                int count=0;
                if(i-1>=0)
                {
                    if(1==board[i-1][j])
                    {
                        count++;
                    }
                    if(j-1>=0)
                    {
                        if(1==board[i-1][j-1])
                        {
                            count++;
                        }
                    }
                    if(j+1<=n-1)
                    {
                        if(1==board[i-1][j+1])
                        {
                            count++;
                        }                           
                    }
                }
                if(i+1<=m-1)
                {
                    if(1==board[i+1][j])
                    {
                        count++;                    
                    }
                    if(j-1>=0)
                    {
                        if(1==board[i+1][j-1])
                        {
                            count++;
                        }
                    }
                    if(j+1<=n-1)
                    {
                        if(1==board[i+1][j+1])
                        {
                            count++;
                        }                           
                    }
                }
                if(j-1>=0 && 1==board[i][j-1])
                {
                    count++;
                }
                if(j+1<=n-1 && 1==board[i][j+1])
                {
                    count++;
                }
                if(1==board[i][j])
                {
                    if(count<2 || count>3)
                    {
                        flag[i][j]=1;
                    }
                }
                else
                {
                    if(3==count)
                    {
                        flag[i][j]=1;
                    }
                }
            }
        }
        for(int i=0;i<m;i++)
        {
            for(int j=0;j<n;j++)
            {
                board[i][j]^=flag[i][j];
            }
        }        
    }
};