289. Game of Life

最新推荐文章于 2023-12-05 10:50:48 发布

原创最新推荐文章于 2023-12-05 10:50:48 发布 · 200 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

LeetCode 专栏收录该内容

28 篇文章

订阅专栏

本文提供了一种解决LeetCode上游戏生命棋问题的方法，通过在原数组上使用二进制位运算技巧，实现原地更新数组状态，避免了额外空间的使用。算法首先遍历数组计算每个元素周围活细胞的数量，再根据生命游戏规则更新每个位置的状态，最后将所有元素的状态右移一位以完成从当前代到下一代的转换。

https://leetcode.com/problems/game-of-life/

1. 思路

使用naive的解法遍历数组内的每个元素，但是改变in-place地改变每个位置的值后，就会影响后面位置的next generation判断。

解决方法是把next generation结果放在二进制数的第二位，不影响当前的状态（0 1都存储在第1位），采用&1操作取得当前的状态，采用>>1操作获得next generation结果

2. AC代码

class Solution {
public:
    void gameOfLife(vector<vector<int>>& board) {
        int m = board.size();
        int n = board[0].size();
        for(int i=0; i<m; i++){
            for(int j=0; j<n; j++){
                int cnt = 0;
                for(int a=max(0, i-1); a<=min(i+1, m-1); a++){
                    for(int b=max(0, j-1); b<=min(j+1, n-1); b++){
                        if(a==i && b==j) continue;
                        if((board[a][b]&1)==1) cnt++;
                    }
                }
                if((board[i][j]&1) == 1 && (cnt==2 or cnt==3)) board[i][j] |= 2;
                if((board[i][j]&1) == 0 && cnt==3) board[i][j] |= 2;
            }
        }
        for(int i=0; i<m; i++){
            for(int j=0; j<n; j++){
                board[i][j] >>= 1;
            }
        }
    }
};