leetcode 765.情侣牵手

leetcode 765.情侣牵手

题干

N 对情侣坐在连续排列的 2N 个座位上，想要牵到对方的手。 计算最少交换座位的次数，以便每对情侣可以并肩坐在一起。 一次交换可选择任意两人，让他们站起来交换座位。
人和座位用 0 到 2N-1 的整数表示，情侣们按顺序编号，第一对是 (0, 1)，第二对是 (2, 3)，以此类推，最后一对是 (2N-2, 2N-1)。
这些情侣的初始座位  row[i] 是由最初始坐在第 i 个座位上的人决定的。

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

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

说明:
len(row) 是偶数且数值在 [4, 60]范围内。
可以保证row 是序列 0…len(row)-1 的一个全排列。

题解

对于每一坨连通在一起的情侣们，将他们变换为相邻位置所需的最少次数即涉及的情侣对数-1。
所以尝试使用并查集，统计一共有多少连通在一起的情侣，比如一共5对情侣，前两对连通在一起，后两队连通在一起，那最少交换次数大概就是两次(2-1)+(2-1)
我们从0开始给每队情侣进行标号，一共有n/2对情侣。
对于第i个人，row[i]/2即为他对应的情侣标号

对于每两个相邻的人，可能有几种情况：
两人恰好是一对情侣： 跳过
两人不是情侣：
两人所在的情侣对此前都未被连通到其他组里： 出现了一个新的组，这个组已经有两对情侣了
两人中的一个所在的情侣对此前被连通到过其他组里 ： 又一对情侣被卷进了之前出现过的一个组
两人所在的情侣对此前都被连通到其他组里： 如果两个组不重合，合并两个组

class Solution {
public:
    int findRoot(int n,vector<int>& ancestor){
        if(ancestor[n] == -1){
            return -1;
        }
        while(ancestor[n] != n){
            n = ancestor[n];
        }
        return n;
    }

    int minSwapsCouples(vector<int>& row) {
        int n = row.size();
		//把每个人预处理为情侣对标号
        for(int i = 0 ; i < n ; ++i){
            row[i] /= 2;
        }
        vector<int> ancestor(n/2+5,-1);
        int ans = 0;
        for(int i = 0 ; i < n - 1 ; i += 2){
            int leftOne = row[i];
            int rightOne = row[i+1];
            int leftRoot = findRoot(leftOne,ancestor);
            int rightRoot = findRoot(rightOne,ancestor);
            if(leftOne == rightOne){
                //如果两个人恰好是情侣
                continue;
            }else if(leftRoot == -1 && rightRoot == -1){
                //如果两人不是情侣且所在的情侣对此前不属于任何组
                //将左边人所在的情侣对作为新的组的祖先
                ancestor[leftOne] = leftOne;
                ancestor[rightOne] = leftOne;
                ans++;
            }else if(leftRoot != -1 && rightRoot == -1){
                //如果左边人所在情侣对已有组，但右边的没有，将右边所属的情侣对加到左边的组
                ancestor[rightOne] = leftRoot;
                ans++;
            }else if(leftRoot == -1 && rightRoot != -1){
                //如果右边人所在的情侣对已有组，但左边的没有，将左边所属的情侣对加到右边的组
                ancestor[leftOne] = rightRoot;
                ans++;
            }else if(leftRoot != -1 && rightRoot != -1 && leftRoot != rightRoot){
                //两人所在的情侣对已有组且是不同组的情况
                //合并两个组，将右边最高祖先的祖先设为左边最高祖先
                //由于合并ans++
                ancestor[rightRoot] = leftRoot;
                ans++;
            }
            //剩下一种情况是两人所属情侣对在同一组的情况，由于不影响答案所以不计入考虑范畴
        }
        return ans;
    }
};