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

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

手写堆版本 dijkstra   400+ms 看来优先队列的常数好大

#include<bits/stdc++.h>
using namespace std;
#define maxn 100001
#define inf 0x3f3f3f3f
#define pii pair<int,int >
inline int read()
{
    ;
    char c=getchar();
    ;
    ;    c=getchar();}
    )+(x<<)+c-';c=getchar();}
    return flag?-x:x;
}
struct ac{
  int to,va,cos,nex;
}eg[maxn*];
struct wa{
   int x,y;
   wa(){}
   wa(int a,int b){
     x=a,y=b;
   }
}aa[maxn*];
int head[maxn],dis[maxn],pre[maxn],flow[maxn],h[maxn];
bool fa[maxn];
,maxcost=,tot=,len=;

void init(){
  tot=;
  memset(eg,,sizeof(eg));
  memset(head,-,sizeof(head));
  memset(h,,sizeof(h));
  memset(pre,,sizeof(pre));
}

bool dcmp(wa A,wa B){
    ;
    ;
}

void add(int v){
    )return;
    ]))swap(aa[v],aa[v/]),add(v/);
}

void updata(int v){
    >len)return;
    ==len)
    {
        ],aa[v]))swap(aa[v],aa[v*]);
        return;
    }
    ])&&dcmp(aa[v],aa[v*+]))return;
    ],aa[v*+]))swap(aa[v],aa[v*]),updata(v*);
    +]),updata(v*+);
}
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=;
   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));
    dis[s]=;
    flow[s]=inf;
    aa[++len]=wa(,s);
    while(len){
      wa x=aa[];
      swap(aa[],aa[len--]);
      updata();
      int u=x.y;
      if(dis[x.y]<x.x) continue;
      ;j=eg[j].nex){
          int v=eg[j].to;
          int f=eg[j].cos;
          &&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;
            aa[++len]=wa(dis[v],v);
            add(len);
          }
      }
    }
    ;
    ;
}
void work(){
   while(dijstra(s,t)){
      int z=t;
      while(z!=s){
         int i=pre[z];
         eg[i].va-=flow[t];
         eg[i^].va+=flow[t];
         z=eg[i^].to;
      }
      ;j<=n;j++){
         h[j]+=dis[j];
      }
      maxflow+=flow[t];
      maxcost+=flow[t]*h[t];
   }
}
int main(){
   cin>>n>>m>>s>>t;
   init();
   ;j<m;j++){
      int x,y,a,b;
      //scanf("%d%d%d%d",&x,&y,&a,&b);
      x=read(); y=read();a=read();b=read();
      add_eg(x,y,a,b);
   }
   work();
   printf("%d %d\n",maxflow,maxcost);
}

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