LintCode 解题记录17.5.13(tag: Union Find)

Union Find - 并查集。
看到这里最先 想到的就是 最小生成树的Kruscal算法利用并查集来实现。脑海里回忆一下思路与代码。

LintCode 178 图是否是树
图是否是树有两个条件：
1.是否形成了环路
2.是否只有一个连通块
自然就会想到常见的求连通块的方法，并查集 || DFS。至于判断是否有环，题目中说明了是无向图。(如果是有向图，又如何判断呢？)
由于题目中说了输入样例中不会都出现(a, b), (b, a)。所以这里我选择用 并查集来 判断是否有回路。
即：对于给的一条边(a, b)，如果发现 a, b 已经属于同一个集合，那么再次加入这条边就一定会形成 环。否则就把a ,b合并到一个集合中。

    int find(vector<int> &parent, int x) {
        if (parent[x] == -1) return x;
        int tmp = find(parent, parent[x]);
        parent[x] = tmp;
        return tmp;
    }

    bool Union(vector<int> &parent, int x, int y) {
        x = find(parent, x);
        y = find(parent, y);
        if (x == y) return false; //x和y已经在同一区间内了，再加就会造成环路
        parent[x] = y;
        return true;
    }
    bool validTree(int n, vector<vector<int>>& edges) {
        // Write your code here
        vector<int> parent(n, -1);
        for (auto edge : edges) {
            bool tag = Union(parent, edge[0], edge[1]);
            if (!tag) return false;//没有环路
        }
        int cnt = 0;
        //只有一个连通块
        for (int i = 0; i < n; i++) {
            if (parent[i] == -1) cnt++;
            if (cnt > 1) return false;
        }
        return true;
    }

LintCode 477 被围绕的区域
即给定二维字符数组，把那些被X包围的O都换成X。由于之前做了一下 Number of Island 这道题，所以对于此题 刚开始也是想先求有关O的连通块，如果发现其中有一个O在边缘上，那么维持原样，否则就把O换成X。可是提交发现 内存溢出了，可能是我开了一个二维vis数组导致的。
后来看了一下网上的思路，他们是从边缘上的O出发，利用BFS或者DFS搜寻，把遍历到的每一个O都换成一个第三方字符Y。
最后再把所有的Y换成O，把所有的O换成X。
代码：
DFS

    void dfs(vector<vector<char> > &board, int x, int y) {
            int dx[] = {0, 0, -1, 1};
            int dy[] = {1, -1, 0, 0};
            board[x][y] = 'Y';
            for (int i = 0; i < 4; i++) {
                int nx = x + dx[i];
                int ny = y + dy[i];
                if (nx < 0 || nx >= board.size() || ny < 0 || ny >= board[0].size())
                    continue;
                if (board[nx][ny] == 'O')
                    dfs(board, nx, ny);
            }
    }
    void surroundedRegions(vector<vector<char> >& board) {
        // Write your code here
        vector<pair<int, int> > posi;
        int row = board.size();
        if (row == 0) return;
        int col = board[0].size();
        for (int i = 0; i < row; i++) {
            for (int j = 0; j < col; j++) {
                if (i == 0 || i == row-1 || j == 0 || j == col-1) {
                    if (board[i][j] == 'O')
                        dfs(board, i, j);
                }
            }
        }
        for (int i = 0; i < row; i++) {
            for (int j = 0; j < col; j++) {
                if (board[i][j] == 'Y')
                    board[i][j] = 'O';
                else if (board[i][j] == 'O')
                    board[i][j] = 'X';
            }
        }
    }

BFS

    void bfs(vector<vector<char> > &board, int x, int y) {
        queue<pair<int, int> > q;
        q.push(make_pair(x, y));
        int dx[] = {0, 0, -1, 1};
        int dy[] = {1, -1, 0 ,0};
        while (!q.empty()) {
            pair<int, int> tmp = q.front();
            q.pop();
            board[tmp.first][tmp.second] = 'Y';
            for (int i = 0; i < 4; i++) {
                int nx = tmp.first + dx[i];
                int ny = tmp.second + dy[i];
                if (nx < 0 || nx >= board.size() || ny < 0 || ny >= board[0].size()) 
                    continue;
                if (board[nx][ny] == 'O') {
                    q.push(make_pair(nx, ny));
                }
            }
        }
    }
    void surroundedRegions(vector<vector<char>>& board) {
        // Write your code here
        int row = board.size();
        if (row == 0) return;
        int col = board[0].size();
        for (int i = 0; i < row; i++) {
            for (int j = 0; j < col; j++) {
                if (i == 0 || i == row-1 || j == 0 || j == col-1) {
                    if (board[i][j] == 'O')
                        bfs(board, i, j);
                }
            }
        }
        for (int i = 0; i < row; i++) {
            for (int j = 0; j < col; j++) {
                if (board[i][j] == 'Y')
                    board[i][j] = 'O';
                else if (board[i][j] == 'O')
                    board[i][j] = 'X';
            }
        }
    }

结果这题为什么要放在 并查集的标签下呢？

总结：
1.并查集 find && Union，以及路径压缩。
2.判断有几个连通块 DFS || Union Find。
3.判断 有向图、无向图 是否有环。
如果对上面的三个问题有点不清晰，或者代码不能很流畅的写出来的， 自行百度学习。