洛谷P4553 80人环游世界

时间:2023-12-31 22:35:14

题目描述

https://www.luogu.org/problemnew/show/P4553

题解

思路比较显然,把图建出来,一个国家拆成两个点,中间设置上下界,然后跑费用流。

我把源那边的流量也设置了上下界,但是题解没有,为什么我按照题解的建图方法会WA?

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<queue>
#define N 209
#define inf 2e9
using namespace std;
queue<int>q;
int head[N],dis[N],tot=,pre[N],fl[N],ans,n,m;
bool vis[N];
inline int rd(){
int x=;char c=getchar();bool f=;
while(!isdigit(c)){if(c=='-')f=;c=getchar();}
while(isdigit(c)){x=(x<<)+(x<<)+(c^);c=getchar();}
return f?-x:x;
}
struct edge{int n,to,l,f;}e[];
inline void add(int u,int v,int l,int f){
e[++tot].n=head[u];e[tot].to=v;head[u]=tot;e[tot].l=l;e[tot].f=f;
e[++tot].n=head[v];e[tot].to=u;head[v]=tot;e[tot].l=;e[tot].f=-f;
}
inline bool spfa(int s,int t){
memset(dis,0x3f,sizeof(dis));
q.push(s);dis[s]=;fl[s]=2e9;
while(!q.empty()){
int u=q.front();q.pop();vis[u]=;
for(int i=head[u];i;i=e[i].n){
int v=e[i].to;
if(e[i].l&&dis[v]>dis[u]+e[i].f){
dis[v]=dis[u]+e[i].f;pre[v]=i;fl[v]=min(fl[u],e[i].l);
if(!vis[v]){vis[v]=;q.push(v);}
}
}
}
return dis[t]!=0x3f3f3f3f;
}
inline void calc(int s,int t){
int x=t;
while(x!=s){
int i=pre[x];
e[i].l-=fl[t];e[i^].l+=fl[t];x=e[i^].to;
}
ans+=dis[t]*fl[t];
}
int main(){
n=rd();m=rd();
int s=*n+,t=*n+,x;
for(int i=;i<=n;++i){
x=rd();add(i,t,x,);add(s,i+n,x,);
add(,s,m,);add(i+n,*n+,m,);
}
// add(2*n+2,0,m,0);
add(s,,m,);add(*n+,t,m,);
add(*n+,*n+,m,);
for(int i=;i<=n;++i)
for(int j=i+;j<=n;++j){
x=rd();if(x>=)add(i+n,j,m,x);
}
while(spfa(s,t))calc(s,t);
printf("%d\n",ans);
return ;
}