leetcode_130.被围绕的区域

给你一个 m x n 的矩阵 board ，由若干字符 ‘X’ 和 ‘O’ ，找到所有被 ‘X’ 围绕的区域，并将这些区域里所有的 ‘O’ 用 ‘X’ 填充。

示例 1：

输入：board = [[“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’。如果两个元素在水平或垂直方向相邻，则称它们是“相连”的。
示例 2：

输入：board = [[“X”]]
输出：[[“X”]]

提示：

m == board.length
n == board[i].length
1 <= m, n <= 200
board[i][j] 为 ‘X’ 或 ‘O’
这个题用并查集非常简单。

class Solution {
public:
    int father[210*210];
    void init() {
        for(int i=0;i<210*210;i++) {
            father[i] = i;
        }
    }
    int findFather(int x) {
        if(x == father[x]) return x;
        father[x] = findFather(father[x]);
        return father[x];
    }
    void Union(int a,int b) {
        int faA = findFather(a);
        int faB = findFather(b);
        if(faA!=faB) {
            father[faA] = faB;
        }
    }
    vector<vector<char>> solve(vector<vector<char>>& board) {
        set<int> mark;
        int m = board.size();
        int n = board[0].size();
        init();
        for(int i=0;i<m;i++) {
            for(int j=0;j<n;j++) {
                if(board[i][j]=='O') {
                    if(i+1<m && board[i+1][j]=='O') {
                        Union(i*n+j,(i+1)*n+j);
                    }
                    if(j+1<n && board[i][j+1]=='O') {
                        Union(i*n+j,i*n+j+1);
                    }
                    if(i-1>=0 && board[i-1][j]=='O') {
                        Union(i*n+j,(i-1)*n+j);
                    }
                    if(j-1>=0 && board[i][j-1]=='O') {
                        Union(i*n+j,i*n+j-1);
                    }
                    //cout<<findFather(i*n+j)<<" ";
                }
            }
        }
        for(int i=0;i<m;i++) {
            for(int j=0;j<n;j++) {
                if(i==0 || j==0 || i==m-1 || j==n-1) {
                    if(board[i][j] == 'O') {
                        int x = findFather(i*n+j);
                        mark.insert(x);
                    }
                }
            }
        }
        set<int>::iterator it;
        for(it=mark.begin();it!=mark.end();it++) {
            //cout<<*it<<" ";
        }
        for(int i=0;i<m;i++) {
            for(int j=0;j<n;j++) {
                if(board[i][j]=='O' && mark.count(findFather(i*n+j))==0) {
                    board[i][j]='X';
                }
            }
        }
        return board;
    }
};