codeforces 400 D Dima and Bacteria【并查集 Floyd】

题意：给出n个点，分别属于k个集合，判断每个集合里面的点的距离都为0，为0的话输出yes,并输出任意两个集合之间的最短路

这道题目有两个地方不会处理，

先是n个点，分别属于k个集合，该怎么记录下来这里，

然后就是判断每个集合里面的点的距离是否为1，这里可以用并查集来做，如果在输入点的时候，距离为0，就将这两点合并

最后判断每个点，如果他们同属于一个集合，判断它俩的根是否一样就可以了

最后用floyd求最短路

 #include<iostream>
 #include<cstdio>
 #include<cstring>
 #include <cmath>
 #include<stack>
 #include<vector>
 #include<map>
 #include<set>
 #include<queue>
 #include<algorithm>
 using namespace std;

 #define foreach(i,c) for (__typeof(c.begin()) i = c.begin(); i != c.end(); ++i)

 typedef long long LL;
 const int INF = (<<)-;
 const int mod=;
 const int maxn=;

 int in[maxn*maxn],d[maxn][maxn],p[maxn*maxn];

 int find(int x){ return x==p[x]? x:p[x]=find(p[x]);}

 int main(){
     int n,m,k;
     scanf("%d %d %d",&n,&m,&k);
     int sum=;
     for(int i=;i<=k;i++){
         int c;
         cin>>c;
         for(int j=sum+;j<=sum+c;j++) in[j]=i;
         sum+=c;
     }

     for(int i=;i<=k;i++){//��ʼ��floyed����
         for(int j=;j<=k;j++){
             if(i==j) d[i][j]=;
             else d[i][j]=INF;
         }
     }
     for(int i=;i<=n;i++) p[i]=i;

     while(m--){
         int u,v,w;
         cin>>u>>v>>w;
         if(in[u]!=in[v]&&d[in[u]][in[v]]>w) d[in[u]][in[v]]=d[in[v]][in[u]]=w;

         if(w==){
             int x=find(u);
             int y=find(v);
             if(x!=y) p[x]=y;
         }
     }

     for(int i=;i<=n;i++){
         if(in[i]==in[i-]){
             int x=find(i);
             int y=find(i-);
             if(x!=y){        //�������ͬһ�����ϵ��Ǹ���ͬ��˵������֮��ķ��ò���0�����NO
                 printf("No\n");
                 return ;
             }
         }
     }
     printf("Yes\n");

     for(int p=;p<=k;p++)
      for(int i=;i<=k;i++)
       for(int j=;j<=k;j++)
       d[i][j]=min(d[i][j],d[i][p]+d[p][j]);

       for(int i=;i<=k;i++){
           for(int j=;j<=k;j++){
               if(d[i][j]==INF) printf("-1 ");
               else printf("%d ",d[i][j]);
             }
           printf("\n");
       }

       return ;
 }