959. 由斜杠划分区域(并查集)

959. 由斜杠划分区域

在由 1 x 1 方格组成的 n x n 网格 grid 中，每个 1 x 1 方块由 '/'、'\' 或空格构成。这些字符会将方块划分为一些共边的区域。

给定网格 grid 表示为一个字符串数组，返回 区域的数量 。

请注意，反斜杠字符是转义的，因此 '\' 用 '\\' 表示。

 

示例 1：

输入：grid = [" /","/ "]
输出：2

示例 2：

输入：grid = [" /","  "]
输出：1

示例 3：

输入：grid = ["/\\","\\/"]
输出：5
解释：回想一下，因为 \ 字符是转义的，所以 "/\\" 表示 /\，而 "\\/" 表示 \/。

 

提示：

• n == grid.length == grid[i].length
• 1 <= n <= 30
• grid[i][j] 是 '/'、'\'、或 ' '

#include <iostream>
#include <vector>
#include <algorithm>
#include <string>

using namespace std;

class UionFind {
public:
    void initUionFind(int n) {
        parent.resize(n);
        count = n;
        for (int i = 0; i < n; i++) {
            parent[i] = i;
        }
    }
    int findRoot(int x) {
        while (x != parent[x]) {
            parent[x] = parent[parent[x]];
            x = parent[x];
        }
        return x;
    }
    void unify(int x, int y) {
        int xRoot = findRoot(x);
        int yRoot = findRoot(y);
        if (xRoot == yRoot) {
            return;
        }
        parent[xRoot] = yRoot;
        count--;
        return;
    }
    int getCount() {
        return count;
    }
private:
    vector<int> parent;
    int count;
};

class Solution : public UionFind {
public:
    int regionsBySlashes(vector<string>& grid) {
        int N = grid.size();
        int n = 4 * N * N; // 将一个小方格划分成4个网格
        initUionFind(n);
        for (int i = 0; i < N; i++) {
            for (int j = 0; j < N; j++) {
                int index = 4 * (i * N + j); // 二维网格转换为一维
                char ch = grid[i][j];
                // 1、方格内合并                
                if (ch == '/') { // 合并0、3,合并1、2
                    unify(index, index + 3);
                    unify(index + 1, index + 2);
                } else if (ch == '\\') { // 合并0、1,合并2、3
                    unify(index, index + 1);
                    unify(index + 2, index + 3);
                } else { // 合并0、1、2、3
                    unify(index, index + 1);
                    unify(index + 1, index + 2);
                    unify(index + 2, index + 3);
                }
                // 2、方格间合并
                if (j + 1 < N) { // 向右合并, 1(当前)、3(右下一列)
                    unify(index + 1, 4 * (i * N + j + 1) + 3);
                }
                if (i + 1 < N) { // 向下合并, 2(当前)、0(下一行)
                    unify(index + 2, 4 * ((i + 1) * N + j));
                }                
            }
        }
        return getCount();
    }
};

int main()
{
    /*
    *   测试用例1:grid = [" /","/ "],期望输出:2
    *   测试用例2:grid = [" /","  "],期望输出:1
    *   测试用例3:grid = ["/\\","\\/"],期望输出:5
    */
    Solution *test = new Solution();
    vector<string> grid = {" /", "/ "};
    int expect = 2;
    int ans = test->regionsBySlashes(grid);
    cout << "expect ans:" << expect << ", actual ans:"<< ans << endl;
    return 0;
}