[LeetCode] 289、生命游戏

题目描述

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

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

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

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

请注意，面板上所有格子需要同时被更新：你不能先更新某些格子，然后使用它们的更新后的值再更新其他格子。

示例：

输入: 
[
  [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更改为另外的值，但是记录其原有的状态和应该转换成的状态。（以便第二次遍历时更改）

• 0——保持0
• 1——保持1
• -2——0转1
• -1——1转0

参考代码

class Solution {
public:
    void gameOfLife(vector<vector<int>>& board) {
        // 暂时先默认输入正确
        int rows = board.size();
        int cols = board[0].size();
        for(int i = 0; i < rows; i++){
            for(int j = 0; j < cols; j++){
                int numOfLift = numOfSurroundingLive(board, i, j, rows, cols);
                if(board[i][j] == 1)
                    board[i][j] = (numOfLift == 2 || numOfLift == 3)? 1: -1;
                else
                    board[i][j] = (numOfLift == 3)? -2: 0;
            }
        }
        
        for(int i = 0; i < rows; i++){
            for(int j = 0; j < cols; j++){
                if(board[i][j] == 0 || board[i][j] == 1)
                    continue;
                    
                if(board[i][j] == -1)
                    board[i][j] = 0;
                else if(board[i][j] == -2)
                    board[i][j] = 1;
            }
        }
        
    }
    
    // 统计某位置细胞的四周的8个位置上活细胞的数目
    int numOfSurroundingLive(vector<vector<int> > &board, int i, int j, int rows, int cols){
        int top = max(0, i-1);
        int bottom = min(rows-1, i+1);
        int left = max(0, j-1);
        int right = min(cols-1, j+1);
        
        int numOfLife = 0;
        for(int x = top; x <= bottom; x++){
            for(int y = left; y <= right; y++){
                if(x == i && y == j)
                    continue;
                if(board[x][y] == 1 || board[x][y] == -1)
                    numOfLife++;
            }
        }
        
        return numOfLife;
    }
    
};