第一次写最小费用最大流

来源：互联网发布：pscc破解补丁mac版编辑：程序博客网时间：2024/05/29 06:58

写了POJ3281...那个题是求最大流的，我直接用它来验证一下最大流求得对不对。。。

#include<iostream>#include<cstdio>#include<cstring>#include<algorithm>#include<climits>#include<queue>#include<mylib.h>#define CLR(arr, val) memset(arr,val,sizeof(arr))using namespace std;const int MAX=200,INF=100000000;class MinCostMaxFlow{public:int cost[MAX][MAX], flow[MAX][MAX], cap[MAX][MAX];int GetMinCostMaxFlow(int s,int t,int v){int max_flow=0;while(1){SPFA(s, v);if(pre[t]==-1) break;int cur=t,minc=INF;do{if(minc>cap[pre[cur]][cur]-flow[pre[cur]][cur]) minc=cap[pre[cur]][cur]-flow[pre[cur]][cur];}while((cur=pre[cur])!=s);max_flow+=minc;cur=t;do{flow[pre[cur]][cur]+=minc;flow[cur][pre[cur]]-=minc;}while((cur=pre[cur])!=s);}return max_flow;}private:int len[MAX],pre[MAX];void SPFA(int s,int v){fill(len,len+v,INF);CLR(pre,-1);len[s]=0;queue<int> q;q.push(s);while(!q.empty()){int u=q.front();q.pop();for(int i=0;i!=v;i++) if(cap[u][i]-flow[u][i] > 0 && len[u]+cost[u][i] < len[i]) {pre[i]=u;len[i] = len[u]+cost[u][i];q.push(i);}}}};MinCostMaxFlow g;int main(){freopen("in.txt","r",stdin);int N,F,D,f,d,n;scanf("%d%d%d",&N,&F,&D);int s=2*N+F+D,t=s+1;fill((int*)g.cost,(int*)g.cost+s+2,1);for(int i=0;i!=N;i++)g.cap[i][i+N]=1;for(int i=0;i!=F;i++)g.cap[s][i+2*N]=1;for(int i=0;i!=D;i++)g.cap[i+2*N+F][t]=1;for(int i=0;i!=N;i++){scanf("%d%d",&f,&d);for(int j=0;j!=f;j++){scanf("%d",&n);g.cap[2*N+n-1][i]=1;}for(int j=0;j!=d;j++){scanf("%d",&n);g.cap[i+N][2*N+n+F-1]=1;}}cout<<g.GetMinCostMaxFlow(s,t,s+2)<<endl;}

忽然发现上面的写法似乎有些问题，因为反向的费用应该设置成-1，不然退流时费用无法回退。