CF 400D - Dima and Bacteria

#include<stdio.h>
#include<string.h>
#define inf 0x3f3f3f3f
const int N=100015,MAX=505;
int type[N],father[N],mp[MAX][MAX];

inline int min(int a,int b){
	return a>b?b:a;
}
int find(int x){
	if(father[x]!=x)
		father[x]=find(father[x]);
	return father[x];
}
void floyd(int n){//ok
	for(int i=1;i<=n;i++)//--nk
		mp[i][i]=0;
	for(int k=1;k<=n;k++)
		for(int i=1;i<=n;i++)
			for(int j=1;j<=n;j++)
				mp[i][j]=min(mp[i][j],mp[i][k]+mp[k][j]);
}

int main(){
	int n,m,K,t,cnt=1;
	scanf("%d %d %d",&n,&m,&K);
	for(int i=1;i<=K;i++){//为每个细菌标识类别
		scanf("%d",&t);
		for(int j=0;j<t;j++)
			type[cnt++]=i;
	}
	for(int i=1;i<=n;i++)
		father[i]=i;
	memset(mp,inf,sizeof mp);
	int u,v,z;
	for(int i=0;i<m;i++){
		scanf("%d %d %d",&u,&v,&z);
		if(z==0){//合并0转换的集合
			u=find(u);v=find(v);
			if(u!=v)
				father[u]=v;
		}
		int x=type[u],y=type[v];
		mp[x][y]=min(mp[x][y],z);
		mp[y][x]=min(mp[y][x],z);
	}
	for(int i=2;i<=n;i++){//类型相同则必连通
		if(type[i]==type[i-1]&&find(i-1)!=find(i)){
			printf("No\n");return 0;
		}
	}
	floyd(K);
	printf("Yes\n");
	for(int i=1;i<=K;i++){
		for(int j=1;j<=K;j++)
			printf("%d%c",mp[i][j]==inf?-1:mp[i][j],j==K?'\n':' ');
	}
	return 0;
}