所谓并查集其实就是找谁和谁是同一个团体,也就是匹配问题,刚开始接触的时候,觉得道很懂,代码不会敲,经过一位学姐的讲解,对并查集有了更进一步的理解。
先来看一个题目:
并查集题目1.0版:
数据结构实验:连通分量个数
Time Limit: 1000 ms Memory Limit: 65536 KiB
Problem Description
在无向图中,如果从顶点vi到顶点vj有路径,则称vi和vj连通。如果图中任意两个顶点之间都连通,则称该图为连通图,
否则,称该图为非连通图,则其中的极大连通子图称为连通分量,这里所谓的极大是指子图中包含的顶点个数极大。
例如:一个无向图有5个顶点,1-3-5是连通的,2是连通的,4是连通的,则这个无向图有3个连通分量。
Input
第一行是一个整数T,表示有T组测试样例(0 < T <= 50)。每个测试样例开始一行包括两个整数N,M,(0 < N <= 20,0 <= M <= 200)
分别代表N个顶点,和M条边。下面的M行,每行有两个整数u,v,顶点u和顶点v相连。
Output
每行一个整数,连通分量个数。
Sample Input
2 3 1 1 2 3 2 3 2 1 2
Sample Output
2 1
Hint
所有有关并查集的题目大部分是在这个题目的基础上改变而来,所以举一反三尤为重要。话不多说先上代码:
#include <stdio.h>
#include <stdlib.h>
int f[100010];
int i, n;
void init()
{
for(i = 1;i <= 10010;i++)
{
f[i] = i;
}
}
int getf(int u)
{
if(f[u] == u)
return u;
f[u] = getf(f[u]);
return f[u];
}
void merge(int u, int v)
{
int t1,t2;
t1 = getf(u);
t2 = getf(v);
if(t1 != t2)
f[t1] = t2;
}
int main()
{
int T, n, m, u, v, sum;
scanf("%d",&T);
while(T--)
{
init();
sum = 0;
scanf("%d %d",&n, &m);
for(i = 0;i < m;i++)
{
scanf("%d %d",&u,&v);
merge(u,v);
}
for(i = 1;i <= n;i++)
{
if(i == f[i])
sum++;
}
printf("%d\n",sum);
}
return 0;
}
分为三个函数,第一个是初始化函数,第二个是找各自的首领的函数,第三个是把它们全部连起来的函数。
并查集题目2.0版:
电影节
Time Limit: 1000 ms Memory Limit: 65536 KiB
Problem Description
某届电影节评选电影,共有两部电影进入最后评选环节,有n名观众,每个人有一次投票的机会,每个人都按照规则投给其中一部电影。为了了解情况,记者随机询问了一些人,一共询问了m次,特别神奇的是,记者每次都询问两个人,而且这两个人都把票投给了同一部电影,观众编号为1~n。
Input
多组输入,每组第一行是两个整数n,m (2 <= n <=100000,0 <= m < n/2),接下来m行数据,表示m次询问,每行数据有两个整数a,b代表观众的编号(1 <= a,b <= n),观众a和观众b投票给了同一部电影,接下来一行是两个整数c,d(1 <= c,d <= n)。
Output
对于每一组输入,输出一行,如果观众c和观众d投票给同一部电影,输出”same”,如果不能确定,输出”not sure”。
Sample Input
5 2 1 2 2 3 1 3 5 2 1 2 3 4 1 4 5 2 1 2 3 4 2 5
Sample Output
same not sure not sure
Hint
这道题目就是在上一个的基础上改进了一步,其实就是增加了一些骚操作,注意细节!!!
#include <stdio.h>
#include <stdlib.h>
int f[100010];
int n;
void init()
{
int i;
for(i = 1;i <= n;i++)
{
f[i] = i;
}
}
int getf(int a)
{
if(f[a] == a)
return a;
f[a] = getf(f[a]);
return f[a];
}
void merge(int a,int b)
{
int t1,t2;
t1 = getf(a);
t2 = getf(b);
if(t1 != t2)
{
f[t1] = t2;
}
}
int main()
{
int m, i, a, b, c, d;
while(~scanf("%d %d",&n,&m))
{
init(); //注意不要忘了初始化
for(i = 0;i < m;i++)
{
scanf("%d %d",&a,&b);
merge(a,b);
}
scanf("%d %d",&c,&d);
if(getf(c) == getf(d)) //判断 c d 的祖先是否一样
printf("same\n");
else printf("not sure\n");
}
return 0;
}