#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;
}
最小费用流
最新推荐文章于 2024-03-24 14:17:34 发布