【leetcode】289 生命游戏(数组,原地算法)

题目链接:https://leetcode-cn.com/problems/game-of-life/

题目描述

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

进阶:

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

思路

类似题73 矩阵置零,采用两次遍历数组的原地算法。
(1)第一次遍历时,标记需要更新的位;将活细胞->死细胞标记为-1,死细胞->活细胞标记为1;
(2)第二次遍历时,将标记为-1的置0,标记为2的置1

复杂度分析

  • 时间复杂度:O(mn)
  • 空间复杂度:O(1)

代码

/*
 * -1表示当前为1,下一次更新为0的细胞
 * 2表示当前为0,下一次更新为1的细胞
 */
class Solution {
public:
    void gameOfLife(vector<vector<int>>& board) {
        if(board.empty() || board[0].empty()) return;
        int rows = board.size(), cols = board[0].size();
        
        // 标记状态
        for (int i = 0; i < rows; ++i) {
            for (int j = 0; j < cols; ++j) {
                int cnt = 0;
                // 统计活细胞数量
                for (int k = i-1; k <= i+1 ; ++k) {
                    for (int l = j-1; l <= j + 1 ; ++l) {
                        if(k<0 || l < 0|| k>=rows || l>=cols || (k==i && l ==j))
                            continue;
                        if(board[k][l] == 1 ||board[k][l] == -1)
                            cnt ++;
                    }
                }
                if (board[i][j] == 1){
                    if(cnt < 2 || cnt > 3)
                        board[i][j] =  -1;
                }
                else if(cnt == 3)
                    board[i][j] = 2;
            }
        }

        // 更新状态
        for (int i = 0; i < rows; ++i) {
            for (int j = 0; j < cols; ++j) {
                if(board[i][j] == -1)
                    board[i][j] = 0;
                else if(board[i][j] ==2)
                    board[i][j] = 1;
            }
        }
    }
};

在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/zjwreal/article/details/95322404