并查集

1.合并集合

题目描述：
一共有 n 个数，编号是 1∼n，最开始每个数各自在一个集合中。现在要进行 m 个操作，操作共有两种：

1. M a b，将编号为 a 和 b 的两个数所在的集合合并，如果两个数已经在同一个集合中，则忽略这个操作；
2. Q a b，询问编号为 a 和 b 的两个数是否在同一个集合中；

第一行输入整数 n 和 m。接下来 m 行，每行包含一个操作指令，指令为 M a b 或 Q a b 中的一种。
对于每个询问指令 Q a b，都要输出一个结果，如果 a 和 b 在同一集合内，则输出 Yes，否则输出 No。
每个结果占一行。
测试样例：

输入:
4 5
M 1 2
M 3 4
Q 1 2
Q 1 3
Q 3 4
输出：
Yes
No
Yes

代码如下：

#include<iostream>
using namespace std;
const int N=100010;
int n,m;
int p[N];
int find(int x)//返回x的祖宗节点+路径压缩
{
    if(p[x]!=x)//如果不是根节点，则继续向上查找
    {
        p[x]=find(p[x]);
    }
    return p[x];
}
int main()
{
    scanf("%d %d",&n,&m);
    for(int i=1; i<=n; i++)
    {
        p[i]=i;
    }
    while(m--)
    {
        char op[2];
        int a,b;
        scanf("%s %d %d",op,&a,&b);
        if(op[0]=='M') p[find(a)]=find(b);//让a的根节点指向b的根节点，即a的根节点为b的根节点的子节点
        else
        {
            if(find(a)==find(b)) puts("Yes");
            else puts("No");
        }
    }
    return 0;
}

2.连通块中点的数量

题目描述：
给定一个包含 n 个点（编号为 1∼n）的无向图，初始时图中没有边。现在要进行 m 个操作，操作共有三种：

1. C a b，在点 a 和点 b 之间连一条边，a 和 b 可能相等；
2. Q1 a b，询问点 a 和点 b 是否在同一个连通块中，a 和 b 可能相等；
3. Q2 a，询问点 a 所在连通块中点的数量。

测试样例：

输入：
5 5
C 1 2
Q1 1 2
Q2 1
C 2 5
Q2 5
输出：
Yes
2
3

代码如下：

#include<iostream>
using namespace std;
const int N=100010;
int n,m;
int p[N], size_1[N];
int find(int x)//返回x的祖宗节点+路径压缩
{
    if(p[x]!=x)//如果不是根节点，则继续向上查找
    {
        p[x]=find(p[x]);
    }
    return p[x];
}
int main()
{
    scanf("%d %d",&n,&m);
    for(int i=1; i<=n; i++)
    {
        p[i]=i;
        size_1[i]=1;
    }
    while(m--)
    {
        char op[5];
        int a,b;
        scanf("%s",op);
        if(op[0]=='C')
        {
            scanf("%d %d",&a,&b);
            if(find(a)==find(b))
            {
                continue;
            }
            size_1[find(b)]=size_1[find(b)]+size_1[find(a)];
            p[find(a)]=find(b);//让a的根节点指向b的根节点，即a的根节点为b的根节点的子节点
        }
        else if(op[1]=='1')
        {
            scanf("%d %d",&a,&b);
            if(find(a)==find(b)) puts("Yes");
            else puts("No");
        }
        else
        {
            scanf("%d",&a);
            printf("%d\n",size_1[find(a)]);
        }
    }
    return 0;
}

例题：7-74 排座位 (25 分)

布置宴席最微妙的事情，就是给前来参宴的各位宾客安排座位。无论如何，总不能把两个死对头排到同一张宴会桌旁！这个艰巨任务现在就交给你，对任何一对客人，请编写程序告诉主人他们是否能被安排同席。

输入格式：
输入第一行给出3个正整数：N（≤100），即前来参宴的宾客总人数，则这些人从1到N编号；M为已知两两宾客之间的关系数；K为查询的条数。随后M行，每行给出一对宾客之间的关系，格式为：宾客1 宾客2 关系，其中关系为1表示是朋友，-1表示是死对头。注意两个人不可能既是朋友又是敌人。最后K行，每行给出一对需要查询的宾客编号。

这里假设朋友的朋友也是朋友。但敌人的敌人并不一定就是朋友，朋友的敌人也不一定是敌人。只有单纯直接的敌对关系才是绝对不能同席的。

输出格式：
对每个查询输出一行结果：如果两位宾客之间是朋友，且没有敌对关系，则输出No problem；如果他们之间并不是朋友，但也不敌对，则输出OK；如果他们之间有敌对，然而也有共同的朋友，则输出OK but…；如果他们之间只有敌对关系，则输出No way。
输入样例：

7 8 4
5 6 1
2 7 -1
1 3 1
3 4 1
6 7 -1
1 2 1
1 4 1
2 3 -1
3 4
5 7
2 3
7 2

输出样例：

No problem
OK
OK but...
No way

#include<iostream>
using namespace std;

int arr[110][110];
int p[110];
int find(int x)
{
    if(p[x]!=x) p[x]=find(p[x]);
    return p[x];
}
int main()
{
    int n,m,k;
    cin>>n>>m>>k;
    for(int i=1;i<=n;i++) p[i]=i;
    for(int i=0;i<m;i++)
    {
        int a,b,c;
        cin>>a>>b>>c;
        if(c==1) p[find(a)]=find(b);
        else
        {
            arr[a][b]=1;
            arr[b][a]=1;
        }
    }
    for(int i=0;i<k;i++)
    {
        int a,b;
        cin>>a>>b;
        if(find(a)==find(b)&&arr[a][b]==0) cout<<"No problem"<<endl;
        else if(find(a)!=find(b)&&arr[a][b]==0) cout<<"OK"<<endl;
        else if(find(a)==find(b)&&arr[a][b]==1) cout<<"OK but..."<<endl;
        else cout<<"No way"<<endl;
    }
    return 0;
}