Leetcode | Surrounded Regions-CSDN博客

本文链接：https://blog.csdn.net/Dora_Yh/article/details/78314200

原题链接：https://leetcode.com/problems/surrounded-regions

题目描述：

Given a 2D board containing ‘X’ and ‘O’ (the letter O), capture all regions surrounded by ‘X’.

A region is captured by flipping all ‘O’s into ‘X’s in that surrounded region.

For example,

X X X X
X O O X
X X O X
X O X X

After running your function, the board should be:

X X X X
X X X X
X X X X
X O X X

题意描述的大概就是我们通常的围棋规则，可以把X和O想成黑白棋，不过这里比较特殊的是，只能是X包围O才可以把O“吃掉”，而O是不能吃掉X的，所以给定一个棋盘，最终是要输出吃掉O之后的棋盘，只有当O被完全包围时才会被全部消灭。

看到这道题时，我首先想到的是用并查集的思想，把里面连在一起的O想成一个连通分量，然后用并查集来进行处理，但又想到如果这个连通分量如果里面出现一个O在棋盘边缘，那么就不用去除整个连通分量，所以想到这里，其实解决这道题其实不用那么复杂，我们可以先从棋盘边缘开始遍历，即从边缘找到O的棋，然后再顺着这个O去寻找与他相邻的所有O，如果有的话就把它变成A，注意这里把它变成A是为了标记它们不用去除的，最后在遍历整个棋盘，把剩下还有的O变成X，然后再将A恢复成O。

算法源代码如下所示：

class Solution {
public:
    void solve(vector<vector<char>>& board) {
        int i,j;
        int row=board.size();
        if(!row)
            return;
        int col=board[0].size();

        for(i=0;i<row;i++){
            check(board,i,0,row,col);
            if(col>1)
                check(board,i,col-1,row,col);
        }
        for(j=1;j+1<col;j++){
            check(board,0,j,row,col);
            if(row>1)
                check(board,row-1,j,row,col);
        }
        for(i=0;i<row;i++)
            for(j=0;j<col;j++)
                if(board[i][j]=='O')
                    board[i][j]='X';
        for(i=0;i<row;i++)
            for(j=0;j<col;j++)
                if(board[i][j]=='1')
                    board[i][j]='O';
    }
    void check(vector<vector<char> >&vec,int i,int j,int row,int col){
        if(vec[i][j]=='O'){
            vec[i][j]='1';
            if(i>1)
                check(vec,i-1,j,row,col);
            if(j>1)
                check(vec,i,j-1,row,col);
            if(i+1<row)
                check(vec,i+1,j,row,col);
            if(j+1<col)
                check(vec,i,j+1,row,col);
        }
    }
};