765. 情侣牵手(并查集)

765. 情侣牵手

n 对情侣坐在连续排列的 2n 个座位上，想要牵到对方的手。

人和座位由一个整数数组 row 表示，其中 row[i] 是坐在第 i 个座位上的人的 ID。情侣们按顺序编号，第一对是 (0, 1)，第二对是 (2, 3)，以此类推，最后一对是 (2n-2, 2n-1)。

返回 最少交换座位的次数，以便每对情侣可以并肩坐在一起。 每次交换可选择任意两人，让他们站起来交换座位。

 

示例 1:

输入: row = [0,2,1,3]
输出: 1
解释: 只需要交换row[1]和row[2]的位置即可。

示例 2:

输入: row = [3,2,0,1]
输出: 0
解释: 无需交换座位，所有的情侣都已经可以手牵手了。

 

提示:

• 2n == row.length
• 2 <= n <= 30
• n 是偶数
• 0 <= row[i] < 2n
• row 中所有元素均无重复

class Solution {
public:
    vector<int> parent;
    int numOfGroup;
    void initUnionFind(int n) {
        parent.resize(n);
        numOfGroup = n;
        for (int i = 0; i < n; i++) {
            parent[i] = i;
        }
    }
    int findRoot(int x) {
        int root = x;
        while (root != parent[root]) {
            root = parent[root];
        }
        while (x != root) {
            int next = parent[x];
            parent[x] = root;
            x = next;
        }
        return root;
    }
    bool isConnected(int x, int y) {
        return (findRoot(x) == findRoot(y));
    }
    void unify(int x, int y) {
        if (isConnected(x, y)) {
            return;
        }
        int xRoot = findRoot(x);
        int yRoot = findRoot(y);
        if (xRoot < yRoot) {
            parent[xRoot] = yRoot;
        } else {
            parent[yRoot] = xRoot;
        }
        numOfGroup--;
        return;
    }
    int minSwapsCouples(vector<int>& row) {
        int length = row.size(); // 座位或者人的个数
        int n = length / 2; // n对情侣
        initUnionFind(n); // 初始化并查集
        for (int i = 0; i < length; i+= 2) {
            unify(row[i] / 2, row[i + 1] / 2);
        }
        // 结果为交换后的情侣对(组数) - 交换前的情侣对(组数)
        return (n - numOfGroup);
    }
};