LeetCode 130. 被围绕的区域

LeetCode 130. 被围绕的区域

题目

给定一个二维的矩阵，包含 ‘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’。如果两个元素在水平或垂直方向相邻，则称它们是“相连”的。

分析

本来想每次遇到O就DFS两次，第一次标记是否在边缘，第二次改变board的值，但是长数据超时。所以直接寻找边缘上的O，如果遇到就DFS一次将其标记为T，然后再遍历一遍board，如果是O则改为X，如果是T则改回O。

C++代码

class Solution {
public:
    void solve(vector<vector<char>>& board) {
        if(board.size() < 1){
            return;
        }
        for(int i = 0; i < board.size(); i++){
            if(board[i][0] == 'O'){
                helper1(i, 0, board);
            }
            if(board[i][board[0].size() - 1] == 'O'){
                helper1(i, board[0].size() - 1, board);
            }
        }
        for(int j = 0; j < board[0].size(); j++){
            if(board[0][j] == 'O'){
                helper1(0, j, board);
            }
            if(board[board.size() - 1][j] == 'O'){
                helper1(board.size() - 1, j, board);
            }
        }
        for(int i = 0; i < board.size(); i++){
            for(int j = 0; j < board[0].size(); j++){
                if(board[i][j] == 'O'){
                    board[i][j] = 'X';
                }else if(board[i][j] == 'T'){
                    board[i][j] = 'O';
                }
            }
        }
    }
    void helper1(int row, int col, vector<vector<char>>& board){
        board[row][col] = 'T';
        if(col != 0){
            if(board[row][col - 1] == 'O'){
                helper1(row, col - 1, board);
            }            
        }
        if(col != board[0].size() - 1){
            if(board[row][col + 1] == 'O'){
                helper1(row, col + 1, board);
            }
        }
        if(row != 0){
            if(board[row - 1][col] == 'O'){
                helper1(row - 1, col, board);
            }
        }
        if(row != board.size() - 1){
            if(board[row + 1][col] == 'O'){
                helper1(row + 1, col, board);
            }
        }
    }
};