POJ 2516 Minimum Cost(费用流 建图)

POJ 2516

题目大意

有N个供给商，M个雇主，K种物品。每个供给商对每种物品的的供给量已知，每个雇主对每种物品的需求量的已知，从不同的供给商输送不同的货物到不同的雇主手上需要不同的花费，又已知从供给商Mj送第kind种货物的单位数量到雇主Ni手上所需的单位花费。
问：供给是否满足需求？若是满足，最小运费是多少？

分析

可以看出k个物品之间是相互独立的，有两种方法：1是把每个点拆成k个点，2是对每个物品分别见图求费用之后相加

建立一个源点和汇点就变成了经典费用流问题了，供应商编号1-m,商店编号m+1-m+n源点编号为0，汇点编号为m+n+1

代码

#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
#include<cstdlib>
#include<queue>
#include<map>
#include<algorithm>
#include<set>
using namespace std;
const int INF=0x7fffffff;
const int MAXN=205;
int N,M,K;
int need[MAXN][MAXN];//need[i][j]表示第i号商店需要j号物品的个数
int sumneed[MAXN];//sumneed[i]表示i号物品需要的总量
int supply[MAXN][MAXN];//supply[i][j]表示第i号供应点库存j号物品的个数
int cost[MAXN][MAXN];
int dis[MAXN];
int vis[MAXN];
int pre[MAXN];//pre[v]保存走向v的边
int ans;
struct Edge
{
      int u;
      int v;
      int cap;
      int cost;
      int next;
}edge[100005];//下标从0开始
int head[MAXN];
int edgecount;
void Init()
{
      memset(head,-1,sizeof(head));
      edgecount=0;
}
void Add_edge(int u,int v,int cap,int cost)
{
      edge[edgecount].u=u;edge[edgecount].v=v;edge[edgecount].cap=cap;
      edge[edgecount].cost=cost;edge[edgecount].next=head[u];
      head[u]=edgecount++;
      edge[edgecount].u=v;edge[edgecount].v=u;edge[edgecount].cap=0;
      edge[edgecount].cost=-cost;edge[edgecount].next=head[v];//注意这里是head[v]
      head[v]=edgecount++;
}
bool Spfa(int s,int t,int n)//0表示无增广路
{
      for(int i=0;i<=n;i++)
      {
            dis[i]=INF;
            vis[i]=0;
            pre[i]=-1;
      }
      dis[s]=0;
      vis[s]=1;
      queue<int>Q;
      Q.push(s);
      while(!Q.empty())
      {
            int u=Q.front();
            Q.pop();
            for(int k=head[u];k!=-1;k=edge[k].next)
            {
                  int v=edge[k].v;
                  int cost=edge[k].cost;
                  if(edge[k].cap && dis[v]>dis[u]+cost)
                  {
                        dis[v]=dis[u]+cost;
                        pre[v]=k;
                        if(!vis[v])
                        {
                              vis[v]=1;
                              Q.push(v);
                        }
                  }
            }
            vis[u]=0;
       }
      if(dis[t]==INF)return 0;
      return 1;
}
int MCMF(int s,int t,int n)//返回最大流
{
      int flow=0;
      int minflow=INF;
      int mincost=0;
      while(Spfa(s,t,n))
      {
            minflow= INF;
            for(int k=pre[t];k!=-1;k=pre[edge[k].u])
            {
                  minflow=min(minflow,edge[k].cap);
            }

            flow+=minflow;
            for(int k=pre[t];k!=-1;k=pre[edge[k].u])
            {
                  edge[k].cap-=minflow;
                  edge[k^1].cap+=minflow;
            }
            mincost+=dis[t]*minflow;
      }
      ans+=mincost;
      return flow;
}
int main()
{
    int a;
    while(scanf("%d%d%d",&N,&M,&K)!=EOF)
    {
          memset(sumneed,0,sizeof(sumneed));
          if(N==0 && M==0 && K==0)break;
          for(int i=M+1;i<=M+N;i++)
                for(int j=1;j<=K;j++)
                {
                      scanf("%d",&need[i][j]);
                      sumneed[j]+=need[i][j];
                }
          for(int i=1;i<=M;i++)
                for(int j=1;j<=K;j++)
                      scanf("%d",&supply[i][j]);
          int flow;
          ans=0;
          int flag=0;
          for(int i=1;i<=K;i++)
          {
                Init();
                for(int x=M+1;x<=M+N;x++)//商店
                {
                      for(int y=1;y<=M;y++)//供应商
                      {
                            scanf("%d",&cost[y][x]);
                            Add_edge(y,x,supply[y][i],cost[y][x]);
                      }
                }
                if(flag==1)continue;
                for(int j=1;j<=M;j++)
                      Add_edge(0,j,supply[j][i],0);
                for(int j=M+1;j<=M+N;j++)
                      Add_edge(j,M+N+1,need[j][i],0);
                flow=MCMF(0,M+N+1,M+N+2);
                if(flow!=sumneed[i])
                {
                      cout<<-1<<endl;
                      flag=1;
                      continue;
                }
          }
          if(flag==0){cout<<ans<<endl;}
    }
    return 0;
}