编程语言
首页 > 编程语言> > 【leetcode】289 生命游戏(数组,原地算法)

【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

复杂度分析

代码

/*
 * -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;
            }
        }
    }
};

在这里插入图片描述

标签:cnt,int,细胞,289,++,算法,board,leetcode,更新
来源: https://blog.csdn.net/zjwreal/article/details/95322404