题目链接:Codeforces - Gang Up
因为有停留这个东西,所以我们可以按照时间分层。
然后分层图上最小费用流即可。
应该比较简单,直接看代码吧。
AC代码:
#pragma GCC optimize("-Ofast","-funroll-all-loops")
#include<bits/stdc++.h>
//#define int long long
using namespace std;
const int inf=0x3f3f3f3f;
const int N=1e5+10,M=4e6+10;
int n,m,k,C,D,s,t,d[N],st[N],vis[N],cost,base=55;
int head[N],nex[M],to[M],w[M],flow[M],tot=1;inline void ade(int a,int b,int c,int d){
to[++tot]=b; nex[tot]=head[a]; w[tot]=d; flow[tot]=c; head[a]=tot;
}
inline void add(int a,int b,int c,int d){ade(a,b,c,d); ade(b,a,0,-d);}
inline int spfa(){
queue<int> q; q.push(s); memset(st,0,sizeof st); memset(d,inf,sizeof d); d[s]=0;
while(q.size()){
int u=q.front(); q.pop(); vis[u]=0;
for(int i=head[u];i;i=nex[i]){
if(flow[i]&&d[to[i]]>d[u]+w[i]){
d[to[i]]=d[u]+w[i];
if(!vis[to[i]]) q.push(to[i]),vis[to[i]]=1;
}
}
}
return d[t]<inf;
}
int dfs(int x,int f){
if(x==t) return cost+=d[t]*f,f;
int fl=0; st[x]=1;
for(int i=head[x];i&&f;i=nex[i]){
if(!st[to[i]]&&flow[i]&&d[to[i]]==d[x]+w[i]){
int mi=dfs(to[i],min(flow[i],f));
flow[i]-=mi,flow[i^1]+=mi,fl+=mi,f-=mi;
}
}
return fl;
}
signed main(){
cin>>n>>m>>k>>C>>D; t=1e4+1;
for(int i=1,x;i<=k;i++) cin>>x,add(s,x,1,0);
for(int i=1,a,b;i<=m;i++){
cin>>a>>b;
for(int j=1;j<n+k;j++){
for(int s=1;s<=k;s++){
add((j-1)*base+a,j*base+b,1,(s*2-1)*D);
add((j-1)*base+b,j*base+a,1,(s*2-1)*D);
}
}
}
for(int i=1;i<=n;i++){
for(int j=1;j<n+k;j++) add(i+(j-1)*base,i+j*base,k,0);
}
for(int i=1;i<=k+n;i++) add((i-1)*base+1,t,k,(i-1)*C);
while(spfa()) dfs(s,inf);
cout<<cost;
return 0;
}