最小花费最大流模板

自己写的最小花费最大流模板

#include
#include
#include
#include
#include
using namespace std;
const int maxn=10010;//点数
const int maxx=100100;//边数，注意为实际上边数的两倍
const int INF=1<<30;
int to[maxx],nex[maxx],cap[maxx],flow[maxx],cost[maxx];
int head[maxn];
int pre[maxn],dis[maxn];
int eid;
void init()
{
    eid=0;
    memset(head,-1,sizeof(head));
}//初始化
void addEdge(int u,int v,int ca,int co)//注意是u到v的有向边，ca是容量，co是花费
{
    to[eid]=v;
    cap[eid]=ca;
    cost[eid]=co;
    flow[eid]=0;
    nex[eid]=head[u];
    head[u]=eid++;

    to[eid]=u;
    cap[eid]=0;
    cost[eid]=-co;
    flow[eid]=0;
    nex[eid]=head[v];
    head[v]=eid++;
}
bool inQ[maxn];
bool spfa(int s,int e,int n)//s点到e点的是否存在最短路，n为点的数量，点可以从0开始，即0到n
{
    queue<int> que;
    memset(inQ,false,sizeof(inQ));
    for(int i=0;i<=n;i++)
        dis[i]=INF;
    dis[s]=0,inQ[s]=true;
    pre[s]=-1;
    que.push(s);
    while(!que.empty())
    {
        int u=que.front();
        que.pop();
        inQ[u]=false;
        for(int i=head[u];~i;i=nex[i])
        {
            int v=to[i];
            if(cap[i]-flow[i]>0&&dis[v]>dis[u]+cost[i])
            {
                dis[v]=dis[u]+cost[i];
                pre[v]=i;//这是为了修改边的容量
                if(!inQ[v])
                    que.push(v),inQ[v]=true;
            }
        }
    }
    return dis[e]!=INF;
}
int minCostMaxFlow(int s,int e,int &minCost,int n)//最小流，s点到e的最小花费最大流，n含义同上
{
    int ans=0;
    while(spfa(s,e,n))
    {
        int mint=INF;
        for(int i=pre[e];~i;i=pre[to[i^1]])
        {
            if(mint>cap[i]-flow[i])
                mint=cap[i]-flow[i];
        }
        ans+=mint;
        for(int i=pre[e];~i;i=pre[to[i^1]])
        {
            flow[i]+=mint;flow[i^1]-=mint;
            minCost+=mint*cost[i];
        }
    }
    return ans;
}