Luogu--3381 【模板】最小费用最大流

题目链接 3381 【模板】最小费用最大流

dijkstra 可能因为常数比较大 t了一个点但是 开o2之后可以轻松水过

dijkstra 版本
#include<bits/stdc++.h>
using namespace std;
#define maxn 100001
#define inf 0x3f3f3f3f
#define pii pair<int,int >
int head[maxn],dis[maxn],pre[maxn],flow[maxn],h[maxn];
struct ac{
  int to,va,cos,nex;
}eg[maxn];
int n,m,s,t,maxflow=0,maxcost=0,tot=0;
void init(){
  tot=0;
  memset(eg,0,sizeof(eg));
  memset(head,-1,sizeof(head));
  memset(h,0,sizeof(h));
  memset(pre,0,sizeof(pre));
}
void add_eg(int u,int to,int va,int cos){
   eg[tot].va=va;
   eg[tot].cos=cos;
   eg[tot].to=to;
   eg[tot].nex=head[u];
   head[u]=tot++;

   eg[tot].va=0;
   eg[tot].cos=-cos;
   eg[tot].to=u;
   eg[tot].nex=head[to];
   head[to]=tot++;
}
bool dijstra(int s,int t){
   memset(dis,inf,sizeof(dis));
   memset(fa,0,sizeof(fa));
   dis[s]=0;
   flow[s]=inf;
   fa[s]=1;
   priority_queue<pii,vector<pii>,greater<pii> >q;
   q.push(make_pair(0,s));
   while(!q.empty()){
      pii x=q.top();
      q.pop();
      int u=x.second;
      if(dis[x.second]<x.first) continue;
      for(int j=head[u];j!=-1;j=eg[j].nex){
         int v=eg[j].to;
         int f=eg[j].cos;
         if(eg[j].va>0&&dis[v]>dis[u]+f+h[u]-h[v]){
            dis[v]=dis[u]+f+h[u]-h[v];
            flow[v]=min(flow[u],eg[j].va);
            pre[v]=j;
            q.push(make_pair(dis[v],v));
         }
      }
   }
   if(dis[t]!=inf) return 1;
   return 0;
}
void work(){
   while(dijstra(s,t)){
      int z=t;
      while(z!=s){
         int i=pre[z];
         eg[i].va-=flow[t];
         eg[i^1].va+=flow[t];
         z=eg[i^1].to;
      }
      for(int j=1;j<=n;j++){
         h[j]+=dis[j];
      }
      maxflow+=flow[t];
      maxcost+=flow[t]*h[t];
   }
}
int main(){
   cin>>n>>m>>s>>t;
   init();
   for(int j=0;j<m;j++){
      int x,y,a,b;
      scanf("%d%d%d%d",&x,&y,&a,&b);
      add_eg(x,y,a,b);
   }
   work();
   cout<<maxflow<<" "<<maxcost<<endl;
}

SPFA版本

// luogu-judger-enable-o2
#include<bits/stdc++.h>
using namespace std;
#define maxn 100001
#define inf 0x3f3f3f3f
int head[maxn],dis[maxn],pre[maxn],flow[maxn];
bool fa[maxn];
struct ac{
  int to,va,cos,nex;
}eg[maxn*2];
int n,m,s,t,maxflow=0,maxcost=0,tot=0;
void init(){
  tot=0;
  memset(eg,0,sizeof(eg));
  memset(head,-1,sizeof(head));
}
void add_eg(int u,int to,int va,int cos){
   eg[tot].va=va;
   eg[tot].cos=cos;
   eg[tot].to=to;
   eg[tot].nex=head[u];
   head[u]=tot++;

   eg[tot].va=0;
   eg[tot].cos=-cos;
   eg[tot].to=u;
   eg[tot].nex=head[to];
   head[to]=tot++;
}
bool spfa(int s,int t){
   memset(dis,inf,sizeof(dis));
   memset(fa,0,sizeof(fa));
   dis[s]=0;
   flow[s]=inf;
   fa[s]=1;
   queue<int>q;
   q.push(s);
   while(!q.empty()){
      int u=q.front();
      q.pop();
      fa[u]=0;
      for(int j=head[u];j!=-1;j=eg[j].nex){
         int v=eg[j].to;
         int f=eg[j].cos;
         //cout<<u<<" "<<v<<endl;
         if(eg[j].va&&dis[u]+f<dis[v]){
            dis[v]=dis[u]+f;
            flow[v]=min(flow[u],eg[j].va);
            pre[v]=j;
            if(!fa[v]){
               fa[v]=1;
               q.push(v);
            }
         }
      }
   }
   if(dis[t]!=inf) return 1;
   return 0;
}
void work(){
   while(spfa(s,t)){
      int z=t;
      while(z!=s){
         int i=pre[z];
         eg[i].va-=flow[t];
         eg[i^1].va+=flow[t];
         z=eg[i^1].to;
      }
      maxflow+=flow[t];
      maxcost+=flow[t]*dis[t];
   }
}
int main(){
   cin>>n>>m>>s>>t;
   init();
   for(int j=0;j<m;j++){
      int x,y,a,b;
      cin>>x>>y>>a>>b;
      add_eg(x,y,a,b);
   }
   work();
   cout<<maxflow<<" "<<maxcost<<endl;
}