对于100%的数据满足N ≤ 150, M ≤ 20 000, 1 ≤ K ≤ 10, Li ≤ 10 000, 保证小智一行一定能够救出皮卡丘。至于为什么K ≤ 10,你可以认为最终在小智的号召下,小智,小霞,小刚,小建,小遥,小胜,小光,艾莉丝,天桐,还有去日本旅游的黑猫警长,一同前去大战火箭队。
#include <cstdio>
#include <cstring>
#include <iostream>
#include <queue>
using namespace std;
queue<int> q;
int n,m,K,S,T,ans,cnt;
int to[100000],next[100000],cost[100000],flow[100000],dis[400],head[400],inq[400],pe[400],pv[400];
int map[400][400];
int rd()
{
int ret=0,f=1; char gc=getchar();
while(gc<‘0‘||gc>‘9‘) {if(gc==‘-‘)f=-f; gc=getchar();}
while(gc>=‘0‘&&gc<=‘9‘) ret=ret*10+gc-‘0‘,gc=getchar();
return ret*f;
}
void add(int a,int b,int c,int d)
{
to[cnt]=b,cost[cnt]=c,flow[cnt]=d,next[cnt]=head[a],head[a]=cnt++;
to[cnt]=a,cost[cnt]=-c,flow[cnt]=0,next[cnt]=head[b],head[b]=cnt++;
}
int bfs()
{
memset(dis,0x3f,sizeof(dis));
q.push(S),dis[S]=0;
int i,u;
while(!q.empty())
{
u=q.front(),q.pop(),inq[u]=0;
for(i=head[u];i!=-1;i=next[i])
{
if(dis[to[i]]>dis[u]+cost[i]&&flow[i])
{
dis[to[i]]=dis[u]+cost[i],pe[to[i]]=i,pv[to[i]]=u;
if(!inq[to[i]]) inq[to[i]]=1,q.push(to[i]);
}
}
}
return dis[T]<0x3f3f3f3f;
}
int main()
{
n=rd(),m=rd(),K=rd();
S=2*n+2,T=2*n+3;
int i,j,k,a,b,c;
memset(head,-1,sizeof(head));
memset(map,0x3f,sizeof(map));
for(i=0;i<=n;i++) map[i][i]=0;
for(i=1;i<=m;i++)
{
a=rd(),b=rd(),c=rd();
map[a][b]=map[b][a]=min(map[a][b],c);
}
for(k=0;k<=n;k++)
{
for(i=0;i<=n;i++)
{
for(j=0;j<=n;j++)
{
if(i>=k||j>=k)
map[i][j]=min(map[i][j],map[i][k]+map[k][j]);
}
}
}
for(i=0;i<=n;i++)
for(j=i+1;j<=n;j++)
if(map[i][j]<0x3f3f3f3f)
add(i+n+1,j,map[i][j],1<<30);
for(i=1;i<=n;i++)
add(i,i+n+1,0,1<<30),add(S,i+n+1,0,1),add(i,T,0,1),add(i+n+1,0,0,K);
add(0,n+1,0,K);
while(bfs())
{
int mf=1<<30;
for(i=T;i!=S;i=pv[i]) mf=min(mf,flow[pe[i]]);
ans+=dis[T]*mf;
for(i=T;i!=S;i=pv[i]) flow[pe[i]]-=mf,flow[pe[i]^1]+=mf;
}
printf("%d",ans);
return 0;
}