BUPT 1501 多彩瓜 并查集

 

多彩瓜 Submit: 1325   Accepted:318 Time Limit: 2000MS  Memory Limit: 65535KDescription
    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<cstdio> #include<iostream> using namespace std; int a[50010]; int p[50010]; void init(int n) { for(int i=0;i<=n;i++) p[i]=i; } int getfather(int x) { if(p[x]!=x) p[x]=getfather(p[x]); return p[x]; } void Union(int a,int b) { int pa=getfather(a); int pb=getfather(b); p[pa]=pb; } int main() { int n,m,k; while(scanf("%d%d",&n,&m)!=EOF) { if(n==0&&m==0) break; init(m); for(int i=1;i<=n;i++) scanf("%d",&a[i]); //printf("111111111/n"); scanf("%d",&k); for(int i=1;i<=k;i++) { int a,b; scanf("%d%d",&a,&b); Union(a,b); } int num=0; int p=-1; for(int i=1;i<=n;i++) { int p2=getfather(a[i]); if(p!=p2) { num++; p=p2; } } printf("%d/n",num); } return 0; }