poj 2516 Minimum Cost (最小费用流)

题意:

给出m个出场地和n个需求地,给出每个出场第能提供k种物品的数量,需求地需要k种物品的数量。在给出k个矩阵a[i][j]表示第j个出场地到第i个需求地的路费。现在问使得所有需求地得到需求的物品所要的最小费用。

题解:

费用流,根据每种物品建一个网络求解对应物品的最小费用,然后求和。

#include<iostream>
#include<math.h>
#include<stdio.h>
#include<algorithm>
#include<string.h>
#include<string>
#include<vector>
#include<queue>
#include<map>
#include<set>
#define B(x) (1<<(x))
using namespace std;
void cmax(int& a,int b){ if(b>a)a=b; }
void cmin(int& a,int b){ if(b<a)a=b; }
typedef long long ll;
const int oo=0x3f3f3f3f;
const ll OO=1LL<<61;
const int MOD=1000007;
const int maxn=302;
const int maxm=5000;
struct EDGE{
    int v,next,c,f,cost;
}E[maxm<<1];
int head[maxn],tol;
int dis[maxn],pre[maxn];
int vis[maxn];

void Init(){
    memset(head,-1,sizeof head);
    tol=0;
}

void add_edge(int u,int v,int f,int cost){
    E[tol].v=v;
    E[tol].c=f;
    E[tol].f=0;
    E[tol].cost=cost;
    E[tol].next=head[u];
    head[u]=tol++;

    E[tol].v=u;
    E[tol].c=0;
    E[tol].f=0;
    E[tol].cost=-cost;
    E[tol].next=head[v];
    head[v]=tol++;
}

bool Spfa(int s,int t){

    for(int i=0;i<=t;i++){
        pre[i]=-1;
        dis[i]=oo;
        vis[i]=0;
    }
    dis[s]=0;
    vis[s]=1;
    queue<int>q;
    q.push(s);
    while(!q.empty()){
        int u=q.front();q.pop();
        vis[u]=0;
        for(int i=head[u];i!=-1;i=E[i].next){
            int v=E[i].v;
            if(E[i].c>E[i].f&&dis[v]>dis[u]+E[i].cost){
                dis[v]=dis[u]+E[i].cost;
                pre[v]=i;
                if(!vis[v]){
                    vis[v]=1;
                    q.push(v);
                }
            }
        }
    }
    return dis[t]!=oo;
}

int min_cost_maxflow(int s,int t,int& cost){
    int flow=0;
    cost=0;
    while(Spfa(s,t)){
        int Min=oo;
        for(int i=pre[t];i!=-1;i=pre[E[i^1].v]){
            if(Min>E[i].c-E[i].f)
                Min=E[i].c-E[i].f;
        }
        for(int i=pre[t];i!=-1;i=pre[E[i^1].v]){
            E[i].f+=Min;
            E[i^1].f-=Min;
            cost+=Min*E[i].cost;
        }
        flow+=Min;
    }
    return flow;
}

int have[maxn][maxn],need[maxn][maxn];
int have_sum[maxn],need_sum[maxn];
int main(){
    //freopen("G:\\read.txt","r",stdin);
    int N,M,K,s,t,w,f;
    while(scanf("%d %d %d",&N,&M,&K)!=EOF){
        if(N==0&&M==0&&K==0)break;
        Init();
        s=N+M+1;
        t=s+1;
        for(int i=0;i<=K;i++)have_sum[i]=need_sum[i]=0;
        for(int i=1;i<=N;i++){
            for(int j=1;j<=K;j++){
                scanf("%d",&need[i][j]);
                need_sum[j]+=need[i][j];
            }
        }
        for(int i=1;i<=M;i++){
            for(int j=1;j<=K;j++){
                scanf("%d",&have[i][j]);
                have_sum[j]+=have[i][j];
            }
        }
        f=0;
        for(int i=1;i<=K;i++){
            if(have_sum[i]<need_sum[i])
                f=1;
        }
        int ans=0,res;
        for(int k=1;k<=K;k++){
            Init();
            for(int i=1;i<=N;i++){
                for(int j=1;j<=M;j++){
                    scanf("%d",&w);
                    add_edge(j,i+M,oo,w);
                }
            }
            for(int i=1;i<=M;i++)add_edge(s,i,have[i][k],0);
            for(int i=1;i<=N;i++)add_edge(i+M,t,need[i][k],0);
            min_cost_maxflow(s,t,res);
            ans+=res;
        }
        if(f) printf("-1\n");
        else printf("%d\n",ans);
    }
    return 0;
}
/*

*/



你可能感兴趣的:(算法,网络,poj)