Description
dalong的舅舅最近到外地旅游时给dalong带回来一种神奇的水果:多彩瓜。这种瓜太漂亮了,它有很多层,每一层都是一种颜色。由于多彩瓜放久了会变坏,dalong决定每天吃一层,可是多彩瓜有好多层,每天吃一层的话还没吃完多彩瓜就坏了,因为多彩瓜颜色很多,dalong把一些很相近的颜色认为是同一种颜色,这样合并颜色后多彩瓜就没有原来这么多层了,dalong就可以在多彩瓜变坏之前吃完它喽~oh,yeah。现在dalong告诉你多彩瓜最初每一层的颜色,颜色之间的关系,你能帮助他算出吃完这些瓜要几天么?注意,如果dalong认为颜色a和颜色b是同一种颜色,颜色b和颜色c是同一种颜色,那么颜色a和颜色c也是同一种颜色。
Input
多组数据测试
对于每组数据
输入的第一行是一个正整数n(1 <= n <= 50000),m(1 <= m <= 100)表示多彩瓜最初有n层,一共有m种不同的颜色。
第二行有n个数,第i个数colori表示多彩瓜第i层的颜色(1 <= colori <= m).
第三行是一个正整数k(0 <= 3000 <= k),表示有k组颜色存在相似认为相同的关系。
后面k行,每行两个正整数a,b(1 <= a, b <= m 且 a != b),表示dalong认为a和b是同一种颜色。
输入的最后是两个0,表示输入结束,这组数据不用处理
Output
对于每组数据,输出dalong需要几天吃完多彩瓜
Sample Input
5 3
1 1 2 3 1
1
1 2
0 0
Sample Output
3
Hint
因为1和2是同一种颜色,那dalong第一天可以吃掉前3层(1,1,2),第二天可以吃掉第四层(3),第三天吃掉最后一层(1)。因此吃完多彩瓜需要3天。
思路:很显然是点的合并,用并查集。
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
const int N=50005,M=105;
int n,m;
int in[N];
int father[N];
int find(int u)
{
while(father[u]>0)
{
u=father[u];
}
return u;
}
void Union(int a,int b)
{
if(father[a]<father[b])
{
father[a]+=father[b];
father[b]=a;
}
else
{
father[b]+=father[a];
father[a]=b;
}
}
int main()
{
while(scanf("%d%d",&n,&m),n!=0||m!=0)
{
memset(father,-1,sizeof(father));
for(int i=1;i<=n;i++)
{
scanf("%d",&in[i]);
}
int tmp;
scanf("%d",&tmp);
for(int i=1;i<=tmp;i++)
{
int u,v;
scanf("%d%d",&u,&v);
if((u=find(u))!=(v=find(v)))
{
Union(u,v);
}
}
int ans=0;
int now=0;
for(int i=1;i<=n;i++)
{
int tmp=find(in[i]);
if(tmp!=now)
{
now=tmp;
ans++;
}
}
printf("%d\n",ans);
}
return 0;
}