Codeforces 400D Dima And Bacteria 暴力+Floyd

题意:n点,m条带权边的无向图,每个顶点属于k类中的某一类,定义合法,同一类中的任意两点存在代价为0的路径,若合法,求出d[i][j]矩阵,表示第i类到第j类的最小代价.
n,m<=1e5,k<=500,wi<=1e4

同一个类有c[i]个点 则要有c[i]*(c[i]-1)/2条权值为0的路径.相同type转化时,不会经过非0边
首先把边权为0两个端点扔到同一个联通分量中,算出该联通分量中每个type的个数,更新该type路径个数即可判断是否合法.O(M) ...其实相同type必须在同一个联通分量中,否则没有代价0路径存在,用并查集即可.

若合法,则以type为定点,type间的最小代价为边建图(因为相同type间代价为0)
然后跑一遍即可 O(K^3)

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> ii;
const int N=2e5+20;
const int M=5e2+20;
const ll inf=2e15;
vector<int> e
;
ll d[M][M];
ll n,m,k,c
,l
,b
,a
;//l[i] type[i]'s left pos
int vis
,id
;
int get(int x)
{
for(int i=k;i>=1;i--)
{
if(x>=l[i])
return i;
}
}
void dfs(int u)
{
vis[u]=1;
a[id[u]]++;
for(int i=0;i<e[u].size();i++)
if(!vis[e[u][i]])
dfs(e[u][i]);
}
bool check()
{
memset(vis,0,sizeof(vis));
memset(b,0,sizeof(b));
for(int i=1;i<=n;i++)
{
if(!vis[i])
{
memset(a,0,sizeof(a));
dfs(i);
for(int j=1;j<=k;j++)
b[j]+=a[j]*(a[j]-1)/2;
}
}
for(int i=1;i<=k;i++)
if(b[i]!=c[i]*(c[i]-1)/2)
return false;
return true;
}
bool floyd()
{
for(int p=1;p<=k;p++)
for(int i=1;i<=k;i++)
for(int j=1;j<=k;j++)
d[i][j]=min(d[i][j],d[i][p]+d[p][j]);
printf("Yes\n");
for(int i=1;i<=k;i++)
{
if(c[i]==1)
d[i][i]=0;
for(int j=1;j<=k;j++)
{
if(d[i][j]>=inf)
d[i][j]=-1;
printf("%I64d%c",d[i][j],j==k?'\n':' ');
}
}
}
int main()
{
while(cin>>n>>m>>k)
{
ll num=0,zero=0,u,v,w;
l[1]=1;
for(int i=1;i<=k;i++)
scanf("%d",&c[i]),num+=c[i]-1,l[i+1]=l[i]+c[i];
for(int i=1;i<=n;i++)
id[i]=get(i),e[i].clear();
for(int i=1;i<=k;i++)
for(int j=1;j<=k;j++)
d[i][j]=inf;
for(int i=0;i<m;i++)
{
scanf("%d%d%I64d",&u,&v,&w);
if(w==0)
e[u].push_back(v),e[v].push_back(u);
u=get(u),v=get(v);
d[u][v]=min(d[u][v],w);
d[v][u]=min(d[v][u],w);
}
if(check()==false)
{
puts("No");
continue;
}
floyd();
}
return 0;
}