loj6011「网络流 24 题」运输问题(费用流)

最小费用最大流/最大费用最大流。本来挺裸的。然而gg了好久。因为原来的求最长路其实有点问题?因为有负权边,所以清dis的时候不能只清-1,会出锅的。要清-inf才行。我现在可能才真正理解了最小费用最大流?每次找一条最小的边,是有可能找到带反向边的增广路的,只有这样才能保证正确性。

#include 
#include 
#include 
using namespace std;
#define inf 0x3f3f3f3f
#define N 210
inline int read(){
    int x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9') x=x*10+ch-'0',ch=getchar();
    return x*f;
}
int n,m,h[N],num=1,dis[N],path[N],T=201,ans=0;
bool inq[N];
struct edge{
    int to,next,w,c;
}data[21000];
inline int add(int x,int y,int w,int c){
    data[++num].to=y;data[num].next=h[x];h[x]=num;data[num].w=w;data[num].c=c;
    data[++num].to=x;data[num].next=h[y];h[y]=num;data[num].w=0;data[num].c=-c;
}
inline bool spfa(){
    queue<int>q;memset(dis,inf,sizeof(dis));memset(path,0,sizeof(path));
    q.push(0);dis[0]=0;inq[0]=1;
    while(!q.empty()){
        int x=q.front();q.pop();inq[x]=0;
        for(int i=h[x];i;i=data[i].next){
            int y=data[i].to;if(!data[i].w) continue;
            if(dis[x]+data[i].cif(!inq[y]) inq[y]=1,q.push(y);
            }
        }
    }return path[T];
}
inline bool spfa1(){
    queue<int>q;memset(dis,128,sizeof(dis));memset(path,0,sizeof(path));
    q.push(0);dis[0]=0;inq[0]=1;
    while(!q.empty()){
        int x=q.front();q.pop();inq[x]=0;
        for(int i=h[x];i;i=data[i].next){
            int y=data[i].to;if(!data[i].w) continue;
            if(dis[x]+data[i].c>dis[y]){
                dis[y]=dis[x]+data[i].c;path[y]=i;
                if(!inq[y]) inq[y]=1,q.push(y);
            }
        }
    }return path[T];
}
int main(){
//  freopen("a.in","r",stdin);
    n=read();m=read();
    for(int i=1;i<=n;++i) add(0,i,read(),0);
    for(int i=1;i<=m;++i) add(i+n,T,read(),0);
    for(int i=1;i<=n;++i)
        for(int j=1;j<=m;++j) add(i,n+j,inf,read());
    while(spfa()){
        int low=inf,now=T;
        while(path[now]) low=min(low,data[path[now]].w),now=data[path[now]^1].to;
        ans+=low*dis[T];now=T;
        while(path[now]) data[path[now]].w-=low,data[path[now]^1].w+=low,now=data[path[now]^1].to;
    }printf("%d\n",ans);ans=0;num=1;
    for(int i=1;i<=n;++i) ++num,swap(data[num].w,data[num^1].w),++num;
    for(int i=1;i<=m;++i) ++num,swap(data[num].w,data[num^1].w),++num;
    for(int i=1;i<=n;++i)
        for(int j=1;j<=m;++j) data[++num].w=inf,data[++num].w=0;
    while(spfa1()){
        int low=inf,now=T;
        while(path[now]) low=min(low,data[path[now]].w),now=data[path[now]^1].to;
        ans+=low*dis[T];now=T;
        while(path[now]) data[path[now]].w-=low,data[path[now]^1].w+=low,now=data[path[now]^1].to;
    }printf("%d\n",ans);return 0;
}

你可能感兴趣的:(其他oj,-----网络流-------,费用流)