并查集之情侣牵手

题目：https://leetcode-cn.com/problems/couples-holding-hands/

LeetCode 765

02 34 15 67

这样的初始座位我们可以看做：首先一队是情侣是属于一个老大的，同坐一个椅子的，我们也认为他是属于一个老大；（即同属一个连通域）

最后输出的结果为n-老大的个数；

#include <bits/stdc++.h>
using namespace std;
//并查集的应用
class UnionFind{
private:
    //记录并查集集合个数
    int count ;
    //存放老大
    vector<int> parent;
public:
    //返回并查集集合个数
    int getCount(){
        return count;
    }

    //构造函数
    UnionFind(int n ){
        this->count = n;
        this->parent = vector<int>  (n,0);
        for(int i =0;i<n;i++){
            //自己的老大是自己
            parent[i] = i;
        }
    }

    //查询老大
    int find(int x ){
        //查到最大的老大为止，这里用了路径压缩法；
        while(x!=parent[x]){
            parent[x] = parent[parent[x]];
            x = parent[x];
        }
        return x;
    }

    //让两个人认同一个老大
    void setUnion(int x,int y){
        int parentx = find(x);
        int parenty = find(y);
        if(parentx == parenty) return;
        else{
            //让其中一个人的老大认另外一个人为老大
            parent[parentx] = parenty;
            count--;
        }
    }
};
int main() {
    int n = 0;
    //n对情侣
    cin >>n;
    int m =2*n;
    //row是目前情侣的位置
    vector<int> row;
    while(m-->0){
        int num ;
        cin >>num;
        row.emplace_back(num);
    }
    //明确：情侣的编号是满足，index1/2 == index2/2;则index1与index2是情侣；
    //建立并查集，初始集合为n;
    UnionFind * couple = new UnionFind(n);
    for(int i = 0;i<row.size();i=i+2){
        couple->setUnion(row[i]/2,row[i+1]/2);
    }
    cout<< n-couple->getCount()<<endl;
}