int pre[200005];
int finds(int k)
{
if(pre[k]==k)return k;
return pre[k]=finds(pre[k]);
}
void join(int x,int y)
{
int fx=finds(x),fy=finds(y);
if(fx!=fy) pre[fx ]=fy;
}
int main()
{
for(int i=1;i<=n;i++)
pre[i]=i;
return 0;
}
并查集入门题 畅通工程(HDU1232)
#include<stdio.h>
int pre[1000];
int finds(int x)
{
int r=x;
while(pre[r ]!=r)
r=pre[r ];
int i=x; int j;
while(i!=r)
{
j=pre[i ];
pre[i ]=r;
i=j;
}
return r;
}
int main()
{
int n,m,p1,p2,i,total,f1,f2;
while(scanf("%d",&n) && n) //读入n,如果n为0,结束
{
total=n-1; //刚开始的时候,有n个城镇,一条路都没有 //那么要修n-1条路才能把它们连起来
for(i=1;i<=n;i++)
pre[i]=i;
scanf("%d",&m); while(m--)
{ //下面这段代码,其实就是join函数,只是稍作改动以适应题目要求
//每读入一条路,看它的端点p1,p2是否已经在一个连通分支里了
scanf("%d %d",&p1,&p2);
f1=finds(p1);
f2=finds(p2);
//如果是不连通的,那么把这两个分支连起来
//分支的总数就减少了1,还需建的路也就减了1
if(f1!=f2)
{
pre[f2 ]=f1;
total--;
}
//如果两点已经连通了,那么这条路只是在图上增加了一个环 //对连通性没有任何影响,无视掉
}
printf("%d\n",total); //最后输出还要修的路条数
}
return 0;
}
食物链(带权并查集)
题目:
动物王国中有三类动物A,B,C,这三类动物的食物链构成了有趣的环形。A吃B, B吃C,C吃A。
现有N个动物,以1-N编号。每个动物都是A,B,C中的一种,但是我们并不知道它到底是哪一种。
有人用两种说法对这N个动物所构成的食物链关系进行描述:
第一种说法是"1 X Y",表示X和Y是同类。
第二种说法是"2 X Y",表示X吃Y。
此人对N个动物,用上述两种说法,一句接一句地说出K句话,这K句话有的是真的,有的是假的。当一句话满足下列三条之一时,这句话就是假话,否则就是真话。
1) 当前的话与前面的某些真的话冲突,就是假话;
2) 当前的话中X或Y比N大,就是假话;
3) 当前的话表示X吃X,就是假话。
你的任务是根据给定的N(1 <= N <= 50,000)和K句话(0 <= K <= 100,000),输出假话的总数。
Input
第一行是两个整数N和K,以一个空格分隔。
以下K行每行是三个正整数 D,X,Y,两数之间用一个空格隔开,其中D表示说法的种类。
若D=1,则表示X和Y是同类。
若D=2,则表示X吃Y。
Output
只有一个整数,表示假话的数目。
Sample Input
100 7
1 101 1
2 1 2
2 2 3
2 3 3
1 1 3
2 3 1
1 5 5
Sample Output
3
分析:
这道题我想了很久很久,看了很多题解。觉得这个最好理解。这是道带权并查集的题,两种操作。有三个阶级,我们默认A>B>C。所以我们设置了一个比N大三倍的数组,目的就是存这个阶级关系。只要在1到n的大于n+1到2n的,同理。这样扩大三倍数组使得这个关系很明确。很好的一道题,能够帮助理解带权并查集。
代码:
#include<stdio.h>
#include<stdlib.h>
#include<math.h>
#include<string.h>
#include<algorithm>
#include<iostream>
#define exp 1e-9
#define PI acos(-1.0)
#define INF 0x3f3f3f3f
using namespace std;
typedef long long LL;
#define lie() {ans++;continue;}
using namespace std;
int ds[160000];
int finds(int a) //查找阶级老大
{
if(ds[a]==a)return a;
return ds[a]=finds(ds[a]);
}
inline void unite(int a,int b) //使得在同一个阶级
{
ds[finds(a)]=finds(b);
}
inline bool same(int a,int b) //判断是否在一个阶级
{
return finds(a)==finds(b);
}
int main()
{
int n,k;
scanf("%d %d",&n,&k);
for(int i=0;i<n*3;i++)
ds[i]=i;
int ans=0;
while(k--)
{
int type,a,b;
scanf("%d %d %d",&type,&a,&b);
a--; b--;
if(a<0 || a>=n || b<0 || b>=n) //输入错误
lie();
if(type==1)
{
if(same(a,b+n) || same(a,b+n*2))
lie();
unite(a,b);
unite(a+n,b+n);
unite(a+n*2,b+n*2);
}
else
{
if(same(a,b) || same(a,b+n*2))
lie();
unite(a,b+n);
unite(a+n,b+n*2);
unite(a+n*2,b);
}
}
printf("%d",ans);
}