【PTA练习】L2-4 部落 （25 分）以例子总结讲解并查集

L2-4 部落 （25 分）

试题链接：https://pintia.cn/problem-sets/976473537405652992/problems/976477039905349643
这次自己总结一下并查集！（题目在后半部分，看题解的直接跳转后半段吧）

---

并查集

并查集是一种维护集合的数据结构，主要含有查找和合并两个操作
合并：将两个集合或元素合并成一个。
查找：查找两个元素是不是在同一个集合中。
其实现通过设立一个father数组，顾名思义就是一个父亲结点数组，记录这一个结点的上一结点（父亲结点），并且父亲结点也在这个集合内。
注意：同一个集合中应该只存在一个根节点，这个根节点应该是这个集合中所有结点的根节点，是整个这个结合的一个标识点。
如下图，几个结点在father中的表示结果：
fa[1]=1;
fa[2]=1;
fa[3]=2;
fa[4]=2;
fa[5]=5;
fa[6]=6;

接下来说并查集中模板步骤：
1.初始化
一开始所有的结点都是独立离散的，那么就要将其的父亲结点设为他们自己来表示自己是独立的点。
代码：

for(int i=0;i<n;i++){
        father[i] = i;
}

2.查找
前面说了一个结点在一个集合中必有一个根节点来表示这个集合，那么我们可以通过递推与递归的方式不断查找父亲结点，从而层层深入找到根节点老祖宗啊！必须注明一点：递推就是比递归要慢！别问我为什么，问就是不跟你讲！哈哈哈
递推代码：

int find(int x){//此函数是返回元素x所在集合的根节点。
    while(x!=fa[therx]){ //当爸爸不是老祖宗结点时候
        x = father[x];//现在我是爸爸，我去找爸爸的爸爸！
    }
    return x;
}

递归代码：

int find(int x){
    if(father[x]==x){	//如果我就是祖宗结点那么就返回祖宗结点
        return x;
    }else{  //如果不是，就判断爸爸是不是祖宗结点
        return find(father[x]);
    }
}

3.合并
将两个集合或元素合并成一个集合，注明此时只有属于两个不同集合的元素或集合才能合并！所以主要分两步：判断这俩人是不是一家的，如果不是那么可以联姻一下子成为一家子对吧。emmm好好讲话：对于两个给定的元素a和b，判断是不是属于一个集合的，如果不是那么就进行第二步骤合并。
代码：

void Union(int a, int b){
    int fathera = find(a);		//查找a的祖宗
    int fatherb = find(b);		//查找b的祖宗
    if(fathera!=fatherb){		//不是一个祖宗的，就可以努力成为一家子啊
        fa[fatherb] = fathera;
    }
}

注意啊！！！不能简单的将这个元素的父亲就设为另外一个元素啊，注意的是这一句啊fa[fatherb] = fathera;，而不是fa[b] = fathera啊！！！！我一开始就错在这里了啊！！！

接下来就是对简单的并查集的查找函数find的优化(下面例题一开始没有ac就是因为没有优化而导致运行超时了！)

优化等于路径压缩，即：
原来需要不断通过查找父节点来找到祖宗结点，此时只需要将这个集合里所有结点的父亲结点都指向祖宗结点即可！
递推代码：

int find(int x){		//此函数是返回元素x所在集合的根节点。
	int a = x;			//x会在下面变成根节点，在此先进行保存。
    while(x!=fa[therx]){	 //当爸爸不是老祖宗结点时候
        x = father[x];		//现在我是爸爸，我去找爸爸的爸爸！
    }
    //此时x为根节点，修改此集合中所有结点父亲结点为根节点
    while(a != father[x]){
			int z = a;           //因为a要背father[a]覆盖，先存一下a的值，用以修改father[a]的值
			a = father[a];    //a要回溯父亲结点
			father[z] = x;    //将原先的结点a的父亲结点改为根节点。
	}
    return x;
}

递归代码：

int find(int x){
    if(father[x]==x){	//如果我就是祖宗结点那么就返回祖宗结点
        return x;
    }else{  //如果不是，就判断爸爸是不是祖宗结点
        return father[x] = find(father[x]); //直接将根节点赋值为当前结点的父亲结点
    }
}

---

题目：

在一个社区里，每个人都有自己的小圈子，还可能同时属于很多不同的朋友圈。我们认为朋友的朋友都算在一个部落里，于是要请你统计一下，在一个给定社区中，到底有多少个互不相交的部落？并且检查任意两个人是否属于同一个部落。
输入格式：
输入在第一行给出一个正整数N（≤10^​4），是已知小圈子的个数。随后N行，每行按下列格式给出一个小圈子里的人：K P[1] P[2] ⋯ P[K]，其中K是小圈子里的人数，P[i]（i=1,⋯,K）是小圈子里每个人的编号。这里所有人的编号从1开始连续编号，最大编号不会超过
10^​4。之后一行给出一个非负整数Q（≤10 ^4），是查询次数。随后Q行，每行给出一对被查询的人的编号。
输出格式：
首先在一行中输出这个社区的总人数、以及互不相交的部落的个数。随后对每一次查询，如果他们属于同一个部落，则在一行中输出Y，否则输出N。
输入样例：

4
3 10 1 2
2 3 4
4 1 5 7 8
3 9 6 4
2
10 5
3 7

输出样例：

10 2
Y
N

大致思路： 思路就是大概的并查集思路，其中存点的时候就有一点是用了set存点，直接去除重复点来计算共有多少结点！易错易超时的地方就是find函数一定要用优化版本的find函数，不然会超时，其他地方没有什么复杂的地方了！
具体代码如下：

#include <iostream>
#include <bits/stdc++.h>
using namespace std;
int fa[10005], flag[10005];
set<int> s;

int find(int x){
    if(fa[x]==x){
        return x;
    }else{
        return fa[x] = find(fa[x]);
    }
}

void Union(int a, int b){
    int fathera = find(a);
    int fatherb = find(b);
    if(fathera!=fatherb){
        fa[fatherb] = fathera;
    }
}

int main()
{
    ios::sync_with_stdio(false);
    int n, num=0, q, q1, q2;
    cin>>n;
    for(int i=0;i<10005;i++){
        fa[i] = i;
        flag[i] = 0;
    }
    for(int i=0;i<n;i++){
        int k, p[10005];
        cin>>k>>p[0];
        s.insert(p[0]);
        for(int j=1;j<k;j++){
            cin>>p[j];
            s.insert(p[j]);
            Union(p[0], p[j]);
        }
    }
    set<int>::iterator it;
    for(it=s.begin();it!=s.end();it++){
        if(flag[find(*it)]==0){
            num++;
            flag[find(*it)] = 1;
        }
    }
    cout<<s.size()<<" "<<num<<endl;
    cin>>q;
    for(int i=0;i<q;i++){
        cin>>q1>>q2;
        if(find(q1)==find(q2)){
            cout<<"Y"<<endl;
        }else{
            cout<<"N"<<endl;
        }
    }
    return 0;
}

之前就做过并查集的题啦，但是感觉看别人的博客很容易就看懂，但是看完看会之后总还是不够清晰，这次决定自己把脑子里的知识梳理一下，总结一下并查集。继续坚持！一起加油！

这两天准备天梯赛又做了一遍，然后把这版的代码记录一下。再看自己的代码，都忘记之前为啥要用set了，记录一下最新版思路还是一样的！可能用set就更加节约一些时间吧，但是关键还在优化并查集部分！`

#include <iostream>
#include <bits/stdc++.h>
using namespace std;

int father[10005];
int flag[10005];
int Find(int x){
    if(father[x]==x){
        return x;
    }else{
        return father[x] = Find(father[x]);
    }
}

void Union(int a, int b){
    int fa = Find(a);
    int fb = Find(b);
    if(fa!=fb){
        father[fa] = fb;
    }
}

int main()
{
    int n, sum=0;
    cin>>n;
    for(int i=0;i<=10005;i++){
        father[i] = i;
    }memset(flag, 0, sizeof(flag));
    for(int i=0;i<n;i++){
        int k;cin>>k;
        int n0;cin>>n0;
        if(flag[n0]==0){
            sum++;flag[n0]=1;
        }
        for(int j=1;j<k;j++){
            int num;
            cin>>num;
            if(flag[num]==0){
                sum++;flag[num]=1;
            }
            Union(n0, num);
        }
    }
    for(int i=0;i<=sum;i++){
        flag[i]=0;
    }
    int ans=0;
    for(int i=1;i<=sum;i++){
        if(flag[Find(i)]==0){
            ans++;
            flag[Find(i)]=1;
        }
    }
    cout<<sum<<" "<<ans<<endl;
    int q;
    cin>>q;
    for(int i=0;i<q;i++){
        int n1, n2;
        cin>>n1>>n2;
        if(Find(n1)==Find(n2)){
            cout<<"Y"<<endl;
        }else{
            cout<<"N"<<endl;
        }
    }
    return 0;
}