题意:N个细菌,分成K种,每种C个,比如5个细菌,分为2种,C分别为2、3,那么1、2一类,3、4、5一类。有m边条u、v、c,u v之间花费为c。如果同一种细菌之间的任意两个花费不为0,输出No;否则Yes,输出不同类别细菌之间 的最短路。
思路:并查集,费用为0的边,合并。最后同一种类别的在一个集合就代表相互间花费为0。最短路,可以用Floyd。
#include<stdio.h>
#include<iostream>
#include<string.h>
#include<string>
#include<stdlib.h>
#include<math.h>
#include<vector>
#include<list>
#include<map>
#include<set>
#include<stack>
#include<queue>
#include<algorithm>
#include<numeric>
#include<functional>
using namespace std;
typedef long long ll;
typedef pair<int,int> pii;
const int maxn = 1e5+5;
int dis[505][505];
int zh[maxn],fen[505];
int pre[maxn];
void init(int n)
{
for(int i = 0; i <= n; i++)
pre[i] = i;
}
int fid(int x)
{
if(pre[x] == x)
return x;
else
return pre[x] = fid(pre[x]);
}
void he(int x,int y)
{
int nx = fid(x);
int ny = fid(y);
if(nx != ny)
pre[nx] = ny;
}
int main(void)
{
int n,m,k;
while(scanf("%d%d%d",&n,&m,&k)!=EOF)
{
init(n);
int now = 1;
fen[0] = 0;
for(int i = 1; i <= k; i++)
{
int t;
scanf("%d",&t);
fen[i] = t + fen[i-1];
while(t--)
zh[now++] = i;
}
memset(dis,0x3f,sizeof dis);
while(m--)
{
int a,b,c;
scanf("%d%d%d",&a,&b,&c);
if(zh[a] != zh[b] && dis[zh[a]][zh[b]] > c)
dis[zh[b]][zh[a]] = dis[zh[a]][zh[b]] = c;
if(c == 0)
he(a,b);
}
int ans = 1;
for(int i = 1; i <= n; i++)
{
int l = fen[i-1]+1;
int r = fen[i];
int flag = fid(l);
for(int i = l; i <= r; i++)
{
if(fid(i) != flag)
{
ans = 0;
break;
}
}
if(!ans) break;
}
if(!ans)
printf("No\n");
else
{
for(int i = 1; i <= k; i++)
dis[i][i] = 0;
for(int kk = 1; kk <= k; kk++)
{
for(int i = 1; i <= k; i++)
{
for(int j = 1; j <= k; j++)
dis[i][j] = min(dis[i][j],dis[i][kk]+dis[kk][j]);
}
}
printf("Yes\n");
for(int i = 1; i <= k; i++)
{
for(int j = 1; j <= k; j++)
{
if(dis[i][j] >= 0x3f3f3f3f)
printf("-1%c",j==k?'\n':' ');
else
printf("%d%c",dis[i][j],j==k?'\n':' ');
}
}
}
}
return 0;
}