HOJ - 2543最小费用流

题目链接：http://acm.hit.edu.cn/hoj/problem/view?id=2543

这个题目挺有意思。

自己扣了一会儿，发现图挺好建，就把（u,v,f,w） 拆成（u,v,f,0）和（u,v,INF,w）就好了。但是在枚举石头时，我想的是二分石头个数，就需要每次重新建图，把边的信息提前保存下来，发现有点麻烦。

看题解说，利用连续最短路算法，每次找一条最小费用路并尽可能多的增广，直到剩余钱数不够再运一个石头为止。就是直接在模板里面改，太方便了。

 #include <iostream>
 #include <cstdio>
 #include <cstring>
 #include <vector>
 #include <queue>
 #include <algorithm>
 using namespace std;
 const int maxn = 2e3;
 const int INF = 1e9;
 int vis[maxn],dist[maxn];
 int tot,head[maxn];
 int pv[maxn],pe[maxn];
 typedef long long ll;
 struct edge
 {
     int to,pre,cap,cost;
 }e[];
 void init()
 {
     tot = ;
     memset(head,-,sizeof(head));
 }
 void add(int from,int to,int cap,int cost)
 {
     e[tot].pre = head[from];
     e[tot].to = to;
     e[tot].cap = cap;
     e[tot].cost = cost;
     head[from] = tot++;
 }
 void addedge(int from,int to,int cap,int cost)
 {
     add(from,to,cap,cost);
     add(to,from,,-cost);
 }
 int n,m;
 ll c,p;
 int ans = ;
 int min_cost_flow(int s,int t,int f,int& max_flow)
 {
     int ret = ;
     while(f>)
     {
         memset(vis,,sizeof(vis));
         for(int i=;i<maxn;i++) dist[i] = INF;
         dist[s] = ;
         queue<int> q;
         q.push(s);
         while(!q.empty())
         {
             int v = q.front(); q.pop();
             vis[v] = ;
             for(int i=head[v];i>=;i=e[i].pre)
             {
                 int to = e[i].to,cap = e[i].cap,cost = e[i].cost;
                 if(cap>&&dist[to]>dist[v]+cost)
                 {
                     pv[to] = v,pe[to] = i;
                     dist[to] = dist[v] + cost;
                     if(!vis[to]) q.push(to);
                     vis[to] = ;
                 }
             }
         }
         if(dist[t]==INF) return ret;
         ///当所有边的流量都流净后，即没有残余网络，返回。
         int d = f;
         for(int v=t;v!=s;v=pv[v])
         {
             d = min(d,e[pe[v]].cap);
         }
         f -= d;
         max_flow += d;
         ret += d*dist[t]; ///走一单位就消耗dist[t]
         for(int v=t;v!=s;v=pv[v])
         {
             e[pe[v]].cap -= d;
             e[pe[v]^].cap += d;
         }

         ///最有意思，在模板里面修改，利用增广路不断去减少钱数
         if(c>=((ll)d*p+(ll)d*dist[t]))
         {
             c -= ((ll)d*p+(ll)d*dist[t]);
             ans += d;
         }
         else
         {
             ans += (c/((ll)dist[t]+(ll)p)); ///算算单独能运几块石头
             break;
         }
     }
     return ret;
 }
 int main()
 {
     int T;scanf("%d",&T);
     while(T--)
     {
         init(); //别忘写
         scanf("%d %d %lld %lld",&n,&m,&c,&p);
         int u,v,f,w;
         for(int i=;i<=m;i++)
         {
             scanf("%d %d %d %d",&u,&v,&f,&w);
             addedge(u,v,f,);
             addedge(v,u,f,);
             addedge(u,v,INF,w);
             addedge(v,u,INF,w);
         }
         int max_flow = ;
         ans = ;
         min_cost_flow(,,INF,max_flow);
         printf("%d\n",ans);
     }
     return ;
 }