題目描述

給定一個二維的矩陣，包含 'X' 和 'O'（字母 O）。

找到所有被 'X' 圍繞的區域，並將這些區域裡所有的 'O' 用 'X' 填充。

示例：

X X X X
X O O X
X X O X
X O X X
執行你的函式後，矩陣變為：
X X X X
X X X X
X X X X
X O X X

解釋：
被圍繞的區間不會存在於邊界上，換句話說，任何邊界上的 'O' 都不會被填充為 'X'。 任何不在邊界上，或不與邊界上的 'O' 相連的 'O' 最終都會被填充為 'X'。如果兩個元素在水平或垂直方向相鄰，則稱它們是“相連”的。
題目連結： https://leetcode-cn.com/problems/surrounded-regions/

思路

題目的解釋說在邊界上的 O 以及與邊界上的 O 相連的 O 都不會被替換成 X，只有不滿足這個條件的 O 會被替換成 X。所以，我們對邊界上的 O 進行 dfs，構造邊界上 O 的連通塊，並把連通塊中的元素設為 Y，表示這是邊界上的 O，不用替換成 X。DFS 完成後，遍歷原來的陣列，如果是 O，就替換成 X；如果是 Y，就替換成 O。程式碼如下：

class Solution {
    int dirs[4][2] = {{0,1},{1,0},{-1,0},{0,-1}};
public:
    void solve(vector<vector<char>>& board) {
        if(board.empty()) return;

        int m = board.size();
        int n = board[0].size();
        for(int i=0; i<m; i++){
            for(int j=0; j<n; j++){
                if((i==0 || i==m-1 || j==0 || j==n-1) && board[i][j]=='O'){
                    dfs(board, i, j);
                }
            }
        }

        for(int i=0; i<m; i++){
            for(int j=0; j<n; j++){
                if(board[i][j]=='Y') board[i][j] = 'O';
                else if(board[i][j]=='O') board[i][j] = 'X';
            }
        }
    }

    void dfs(vector<vector<char>>& board, int r, int c){
        board[r][c] = 'Y'; // 別忘了這一步
        for(int i=0; i<4; i++){
            int nr = r+dirs[i][0];
            int nc = c+dirs[i][1];
            if(nr>=0 && nr<board.size() && nc>=0 && nc<board[0].size() && board[nr][nc]=='O'){
                board[nr][nc] = 'Y';
                dfs(board, nr, nc);
            }
        }
    }
};