LA 5095 Transportation

1.题目描述：点击打开链接

2.解题思路：本题是指定固定流量的最小费用问题。但是一个难点是，代价不是固定的，而是和流量的平方成正比。可以进行如下的等效转换：设u,v之间的费用的系数为a,最大容量为c,那么可以在u,v之间添加c条边，每条边的容量都是1，费用分别为1*a,3*a,5*a,..(2c-1)*a。这样既可转化为普通的费用流问题。为什么可以这么做呢？因为每个容量都可以用这c条边组合出来。从数学方面解释是因为利用了恒等式1+3+5+...+(2c-1)=c^2。

3.代码：

#include<iostream>
#include<algorithm>
#include<cassert>
#include<string>
#include<sstream>
#include<set>
#include<bitset>
#include<vector>
#include<stack>
#include<map>
#include<queue>
#include<deque>
#include<cstdlib>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<ctime>
#include<cctype>
#include<list>
#include<complex>
#include<functional>
using namespace std;

#define me(s)  memset(s,0,sizeof(s))
#define rep(i,n) for(int i=0;i<(n);i++)
#define pb push_back
typedef long long ll;
typedef unsigned int uint;
typedef unsigned long long ull;
typedef pair <int,int> P;

struct Edge{int to,cap,cost,rev;};
int V;
const int maxv=100+10;
const int INF=1e9;
vector<Edge>g[maxv];
int dist[maxv];
int prevv[maxv],preve[maxv];

void addedge(int from,int to,int cap,int cost)
{
    g[from].pb(Edge{to,cap,cost,g[to].size()});
    g[to].pb(Edge{from,0,-cost,g[from].size()-1});
}

int min_cost_flow(int s,int t,int f)
{
    int res=0;
    while(f>0)
    {
        fill(dist,dist+V,INF);
        dist[s]=0;
        bool update=true;
        while(update)
        {
            update=false;
            for(int v=0;v<V;v++)
            {
                if(dist[v]==INF)continue;
                for(int i=0;i<g[v].size();i++)
                {
                    Edge&e=g[v][i];
                    if(e.cap>0&&dist[e.to]>dist[v]+e.cost)
                    {
                        dist[e.to]=dist[v]+e.cost;
                        prevv[e.to]=v;
                        preve[e.to]=i;
                        update=true;
                    }
                }
            }
        }
        if(dist[t]==INF)return -1;
        int d=f;
        for(int v=t;v!=s;v=prevv[v])
            d=min(d,g[prevv[v]][preve[v]].cap);
        f-=d;
        res+=d*dist[t];
        for(int v=t;v!=s;v=prevv[v])
        {
            Edge&e=g[prevv[v]][preve[v]];
            e.cap-=d;
            g[v][e.rev].cap+=d;
        }
    }
    return res;
}


int main()
{
    int n,m,k;
    while(~scanf("%d%d%d",&n,&m,&k))
    {
        me(g);
        V=n;
        int u,v,a,c;
        for(int i=0;i<m;i++)
        {
            scanf("%d%d%d%d",&u,&v,&a,&c);
            u--,v--;
            for(int j=0;j<c;j++)
                addedge(u,v,1,(2*j+1)*a);//拆边处理
        }
        int flow;
        int ans=min_cost_flow(0,n-1,k);
        if(ans<0)puts("-1");
        else printf("%d\n",ans);
    }
}