BUPT 1501 多彩瓜并查集

    技术2024-08-18  67

     

    多彩瓜 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 Input5 31 1 2 3 111 20 0Sample Output3Hint    因为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; }  

    最新回复(0)