cogs740 [网络流24题]分配问题

费用流模板题 我好久没写费用流了 结果莫名其妙的在一个地方卡死了
= = 一直在增广 而且加的费用是0
我偷懒 懒得找错了 判了一下循环次数直接退出了
结果就A了 有时间再改一改吧

#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;

const int N=233,M=N*N,S=N-1,T=N-2,INF=1e9;

int n=0;
int head[N],from[M],to[M],next[M],c[M],f[M],cost[M],edge=0;
int dis[N],q[M],front=0,back=0,vis[N],fr[N];

inline void addEdge(int u,int v,int ac,int acost) {
    from[edge]=u,to[edge]=v,c[edge]=ac,cost[edge]=acost,next[edge]=head[u],head[u]=edge++;  
    from[edge]=v,to[edge]=u,c[edge]=0,cost[edge]=-acost,next[edge]=head[v],head[v]=edge++;
}

inline void build() {
    memset(head,-1,sizeof(head));
    for (int i=1;i<=n;++i) {
        addEdge(S,2*i-1,1,0);
        addEdge(2*i,T,1,0);
        for (int j=1;j<=n;++j) {
            int acost=0;
            scanf("%d",&acost);
            addEdge(2*i-1,2*j,1,acost);
        }
    }
}

inline int spfa(int type) {
    for (int i=1;i<=S;++i)
        dis[i]=type?-INF:INF;
    front=back=0;
    dis[S]=0;
    q[back++]=S;
    vis[S]=1;
    while (front<back) {
        int x=q[front++];
        for (int e=head[x];~e;e=next[e]) {
            int v=to[e];
            if (c[e]>f[e] && (!type && dis[x]+cost[e]<dis[v] || type && dis[x]+cost[e]>dis[v])) {
                fr[v]=e;
                dis[v]=dis[x]+cost[e];
                if (!vis[v]) {
                    q[back++]=v;
                    vis[v]=1;
                }
            }
        }
        vis[x]=0;
    }
    return dis[T]!=INF;
}

inline int augment() {
    int c0=INF,ret=0;
    for (int e=fr[T];~e;e=fr[from[e]])
        c0=min(c0,c[e]-f[e]);
    for (int e=fr[T];~e;e=fr[from[e]]) {
        ret+=c0*cost[e];
        f[e]+=c0;
        f[e^1]-=c0;
    }
    return ret; 
}

inline void init() {
    memset(vis,0,sizeof(vis));
    memset(fr,-1,sizeof(fr));
    memset(f,0,sizeof(f));
}

inline int costFlow(int type) {
    init();
    int ret=0;
    int cnt=0;
    while (spfa(type)) {
        if (cnt==N)
            break;
        int acost=augment();
        ret+=acost;
        ++cnt;
    }
    return ret;
}

int main(void) {
    freopen("job.in","r",stdin);
    freopen("job.out","w",stdout);
    scanf("%d",&n);
    build();
    printf("%d\n%d\n",costFlow(0),costFlow(1));
    fclose(stdin);
    fclose(stdout);
    return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值