Leetcode_graph 765. 情侣牵手_并查集

 

 

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]的位置即可。

 

1、进入正题：题目给出情侣关系是 (0,1) (2,3) (4,5) (2N-2,2N-1)
通过每个数字除2可以得到 (0,0) (1,1) (2,2) ....
把每个处理后的数字看成一个节点的话，(0,0)构成一个自己指向自己的环，环的概念很关键，新增的一条边的2个节点只要在一个集合里，一定会成环；
那大环是什么样的？为更形象的理解我们这样表达(0-0) (1-1) (2-2)，每个扩号内是一个线段(0-1) (1-2) (2-0) 构成了大环

2、如果K对情侣坐错了位置，需要调整K-1次就可以恢复，比如（除2以前的样子）(0,2) (3,1) 需要2和1换一下位子即可 (0,1) (3,2)
转换成除2后的样子是：(0,1) (1,0) 也是调整1次(0,0) (1,1)

3、那么我们的问题变成了多少对情侣做错了位置，如果K，那么我们只需要调整K-1次

class Solution {
public:
    vector<int> parent,size;
    int findroot(int x){ //套路代码找根，顺带把根的孩子都挂到根下
        return x==parent[x]?x:parent[x]=findroot(parent[x]);
    }


    void uni(int x,int y){ //套路代码——合并
        int xroot = findroot(x);
        int yroot = findroot(y);
        if(xroot != yroot){
        if(size[xroot] < size[yroot]){
            parent[xroot] = yroot;
            size[yroot] += size[xroot];
        }else{
            parent[yroot] = xroot;
            size[xroot] += size[yroot];
        }
        }
    }
    int minSwapsCouples(vector<int>& row) {
        int num = row.size();
        int N = num / 2;
        int circle = 0;
        parent = vector<int>(N,0);
        size = vector<int>(N,1);
        for(int i = 0 ; i < N ; i ++){
            parent[i] = i;
        }
        for(int i = 0 ; i < num ; i += 2){
            uni(row[i]/2,row[i + 1]/2);
        }
            //小环(0-0) (1-1)小环不统计 (0-2) (3-4) (5-1)-->(0-1) (1-2) (2-0)这种大环根不统计
            //正好等于K对情侣坐错了，要调整K-1，统计的circle就是K-1的答案；
        for(int i = 0; i < N ; i ++){
            if(i != findroot(i)){
            circle ++;
            }
        }
        return circle;
    }
};