1. 程式人生 > >poj 2516 Minimum Cost【最大流最小費用流】

poj 2516 Minimum Cost【最大流最小費用流】

題目連結:http://poj.org/problem?id=2516

題目大意:有 n 個商店,m 個倉庫,k 種商品,問滿足供貨商要求的最小花費是多少,如果不能滿足輸出-1;前 n 行為每個商店需要商品的數量,接著 m 行為每個倉庫能放的多少個每種商品,然後是 k 個 n*m 的矩陣,表示第 k 個物品從倉庫 i 到商店 j 的花費:

思路:每種商品求一下最小費用流,求k次;

我的程式碼源點是 0 ,匯點是 n+m+1,與源點連的是商店,與匯點相連的是倉庫;

#include<cstring>
#include<string>
#include<cstdio>
#include<stdlib.h>
#include<iostream>
#include<algorithm>
#include<math.h>
#include<map>
#include<vector>
#include<stack>
#define inf 0x3f3f3f3f
#include<queue>
#include<set>
using namespace std;
typedef long long ll;
const int N=105;
const int M=20005;

//建圖
struct node
{
    int v,f,cos,ne;
} edge[M];
int head[N*2],pre[N*2],dist[N*2],vis[N*2];
int s,t,e;

//讀資料
int ned[N][N],has[N][N],val[N][N][N];
int nsum[N],hsum[N];

void add_insert(int a,int b,int c,int d)
{
    edge[e].v=b;
    edge[e].f=c;
    edge[e].cos=d;
    edge[e].ne=head[a];
    head[a]=e++;
}

void add(int a,int b,int c,int d)
{
    add_insert(a,b,c,d);
    add_insert(b,a,0,-d);
}

int spaf()
{
    memset(pre,-1,sizeof(pre));
    memset(dist,inf,sizeof(dist));
    memset(vis,0,sizeof(vis));
    queue<int>que;
    vis[s]=1;
    dist[s]=0;
    que.push(s);
    while(!que.empty())
    {
        int u=que.front();
        que.pop();
        vis[u]=0;
        for(int i=head[u]; i!=-1; i=edge[i].ne)
        {
            int v=edge[i].v,f=edge[i].f,c=edge[i].cos;
            if(dist[v]>dist[u]+c&&f)
            {
                dist[v]=dist[u]+c;
                pre[v]=i;
                if(!vis[v])
                {
                    vis[v]=1;
                    que.push(v);
                }
            }
        }
    }
    return pre[t]!=-1;
}

int MCMF()
{
    int flow=0,cost=0;
    while(spaf())
    {
        int mini=inf;
        for(int i=pre[t]; i!=-1; i=pre[edge[i^1].v])
            mini=min(mini,edge[i].f);
        for(int i=pre[t]; i!=-1; i=pre[edge[i^1].v])
        {
            edge[i].f-=mini;
            edge[i^1].f+=mini;
            cost+=edge[i].cos*mini;
        }
        flow+=mini;
    }
    return cost;
}

int main()
{
    int n,m,k;
    while(~scanf("%d %d %d",&n,&m,&k))
    {
        if(!n&&!m&&!k) break;
        s=0;t=n+m+1;
        memset(nsum,0,sizeof(nsum));
        memset(hsum,0,sizeof(hsum));
        int ans=0;
        for(int i=1;i<=n;i++)
        {
            for(int j=1;j<=k;j++)
            {
                scanf("%d",&ned[i][j]);//記錄商店需要的貨物的數量
                nsum[j]+=ned[i][j];
            }
        }
        for(int i=1;i<=m;i++)
        {
            for(int j=1;j<=k;j++)
            {
                scanf("%d",&has[i][j]);//記錄倉庫存放的貨物的數量
                hsum[j]+=has[i][j];
            }
        }
        for(int i=1;i<=k;i++)
        {
            for(int j=1;j<=n;j++)
            {
                for(int l=1;l<=m;l++)
                    scanf("%d",&val[i][j][l]);//記錄第l種貨物從i到j的花費
            }
        }
        int flag=1;
        for(int i=1;i<=k;i++)
        {
            if(nsum[i]>hsum[i])
            {
                flag=0;
                break;
            }
        }
        if(!flag)
        {
            printf("-1\n");
            continue;
        }
        for(int l=1;l<=k;l++)
        {
            memset(head,-1,sizeof(head));
            e=0;
            //源點到商店
            for(int i=1;i<=n;i++)
                add(s,i,ned[i][l],0);
            //倉庫到匯點
            for(int i=1;i<=m;i++)
                add(n+i,t,has[i][l],0);
            //商店到倉庫
            for(int i=1;i<=n;i++)
                for(int j=1;j<=m;j++)
                add(i,n+j,min(ned[i][l],has[j][l]),val[l][i][j]);
            ans+=MCMF();
        }
        printf("%d\n",ans);
    }
    return 0;
}