LeetCode 289 生命游戏

题目描述

方法一：使用额外数组
这里主要的技巧就是如何找到一个点周围的八个点，使用了一个8行2列的数组。

class Solution {
    int d[8][2] = {{0,-1},{-1,0},{0,1},{1,0},{-1,-1},{-1,1},{1,1},{1,-1}};
    int m;
    int n;
    bool isArea(int x, int y){
        return x >= 0 && x < m && y >= 0 && y < n;
    }
public:
    void gameOfLife(vector<vector<int>>& board) {
        m = board.size();
        if(m == 0) return;
        n = board[0].size();
        vector<vector<int>> res = board;
        for(int i = 0; i < m; i++){
            for(int j = 0; j < n; j++){
                //计算周围活细胞的数目
                int countLive = 0;
                for(int k = 0; k < 8; k++){
                        int newX = i + d[k][0];
                        int newY = j + d[k][1];
                        if(isArea(newX,newY) && board[newX][newY] == 1){
                            countLive++;
                        }
                }
                if(board[i][j] == 1 && (countLive < 2 || countLive > 3)){ //活细胞
                    res[i][j] = 0; 
                }else if(board[i][j] == 0 && countLive == 3){ //死细胞
                    res[i][j] = 1;
                }
            }
        }
        board = res;
    }
};

方法二：不使用额外数组
题目要求使用原地算法，因为每个位置只有0或1的状态，可以使用当前最低的位保存原始的状态，倒数第二位保存更新后的状态。

class Solution {
    int d[8][2] = {{0,-1},{-1,0},{0,1},{1,0},{-1,-1},{-1,1},{1,1},{1,-1}};
    int m;
    int n;
    bool isArea(int x, int y){
        return x >= 0 && x < m && y >= 0 && y < n;
    }
public:
    void gameOfLife(vector<vector<int>>& board) {
        m = board.size();
        if(m == 0) return;
        n = board[0].size();
        for(int i = 0; i < m; i++){
            for(int j = 0; j < n; j++){
                //计算周围活细胞的数目
                //原有的最低位存储的是当前的状态，倒数第二位存储下一个状态
                int countLive = 0;
                for(int k = 0; k < 8; k++){
                        int newX = i + d[k][0];
                        int newY = j + d[k][1];
                        if(isArea(newX,newY)){
                            countLive += board[newX][newY] & 1;
                        }
                }
                if((board[i][j]&1) > 0){ //活细胞
                    if(countLive >=2 && countLive <=3){
                        board[i][j] = 0b11; //下一个状态还是活细胞
                    }else{
                        board[i][j] = 0b01; //下一个状态是死细胞
                    }
                }else if(countLive == 3){ //死细胞
                    board[i][j] = 0b10; //下一个状态是活细胞
                }
            }
        }
         //最后一位去掉
        for(int i = 0; i < m; i++){
            for(int j = 0; j < n; j++){
                board[i][j] >>= 1;
                }
        }
    }
};

相似题目矩阵置0