最小费用流

#include <iostream>
#include <vector>
#include <queue>
#define MAX 5000
#define INF 999999
using namespace std;
typedef pair<int,int> p;//first最短距离 second 顶点编号
struct edge{int to,cap,cost,rev};//rev为反向边
int v;
vector<edge> G[MAX];
int h[MAX];
int dist[MAX];
int prevv[MAX],preve[MAX];//prevv前驱结点，preve对应的结点
void add_edge(int from,int to,int cap,int cost)
{
    
    G[from].push_back((edge){to,cap,cost,G[to].size()});//G[to].size()表示tp顶点连的边的个数为什么要这样？
    G[to].push_back((edge){from,cap,cost,G[to}.size()-1});
}
//从s到t的流量为f的最小费用流
int min_cost_flow(int s,int t,int f)
{
    int res=0;
    memset(h,0,sizeof(h));
    while(f>0)
    {
        priority_queue<p,vector<p>,greater<p> >q;
        for(int i=0;i<v;i++) dist[i]=INF;
        dist[s]=0;
        q.push(p(0,s));
        while(!q.empty())
        {
            p temp=q.top();
            q.pop();
            int v=temp.second;//v是中转点，通过这个点来进行松弛
            if(dist[v]<temp.first) continue;
            
            
            for(int i=0;i<G[v].size();i++)//G[v].size()表示和v相连的顶点
            {
                edge &e=G[v][i];
                if(e.cap>0&&dist[e.to]>dist[v]+e.cost+h[v]-h[e.to])
                {
                    dist[e.to]=dist[v]+e.cost;
                    prevv[e.to]=v;//前驱结点
                    preve[e.to]=i;//对应的结点
                    q.push(p(dist[e.to],e.to));
                }
            }
        }
        if(dist[t]==INF) return -1;
        for(int i=0;i<v;i++) h[i]=dist[i];
        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*h[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;
}