最小费用最大流模板 Minimum Cost POJ - 2516

题目链接：https://vjudge.net/problem/POJ-2516
题意：输入n,m,k代表商店数、供应商数、商品数。接下来n行表示商店所需的商品情况，m行表示供应商的存货，k个n*m的矩阵代表第i种商品第k个商店从第j个供应商拿去的运输单价，求最小花费。
思路：本来想要一个一个的拆点，觉得太麻烦。参考了https://www.cnblogs.com/kangkang-/p/11329128.html。可以不用拆点，方法是每种商品分别建图。

#include <cstdio>
#include <algorithm>
#include <queue>
#include <vector>
#include <cstring>
#include <cmath>
using namespace std;
const int maxn=1e5+5;
const int inf=1e9;
struct Edge
{
    int from,to,cap,flow,cost;// 起点 终点 容量 流量 费用  最大流令费用=0即可
    Edge(int u,int v,int c,int f,int w):from(u),to(v),cap(c),flow(f),cost(w) {}
};
struct MCMF
{
    int n,m;
    vector <Edge>edges;
    vector <int> g[maxn];
    int inq[maxn];
    int d[maxn];
    int p[maxn];
    int a[maxn];
    void init(int n)
    {
        this->n=n;
        for(int i=0; i<=n; i++)//注意此处的<=
        {
            g[i].clear();
        }
        edges.clear();
    }
    void addedge(int from,int to,int cap,int cost)
    {
        edges.push_back(Edge(from,to,cap,0,cost));
        edges.push_back(Edge(to,from,0,0,-cost));
        m=edges.size();
        g[from].push_back(m-2);
        g[to].push_back(m-1);
    }
    bool spfa(int s,int t,int &flow,long long &cost)
    {
        for(int i=0; i<=n; i++)//此处的<=
        {
            d[i]=inf;
        }
        memset(inq,0,sizeof(inq));
        d[s]=0;
        inq[s]=1;
        p[s]=0;
        a[s]=inf;
        queue<int> q;
        q.push(s);
        while(!q.empty())
        {
            int u=q.front();
            q.pop();
            inq[u]=0;//注意此处 inq数组只标记了在队列内部的 即一条增广路一定不是环
            for(int i=0; i<g[u].size(); i++)
            {
                Edge &e=edges[g[u][i]];
                if(e.cap>e.flow&&d[e.to]>d[u]+e.cost)//容量可以容纳  寻找最小费用
                {
                    d[e.to]=d[u]+e.cost;
                    p[e.to]=g[u][i];//保存路径 为修正边上的流量做准备
                    a[e.to]=min(a[u],e.cap-e.flow);//一条增广路最大可行容量以路径上容量最小的边为准
                    if(!inq[e.to])
                    {
                        q.push(e.to);
                        inq[e.to]=1;
                    }
                }
            }
        }
        if(d[t]==inf)
            return false;
        flow+=a[t];
        cost+=(long long) d[t]*(long long )a[t];
        for(int u=t; u!=s; u=edges[p[u]].from)//修正正、反两条边的容量
        {
            edges[p[u]].flow+=a[t];
            edges[p[u]^1].flow-=a[t];
        }
        return true;
    }
    int mincostmaxflow(int s,int t,long long &cost)
    {
        int flow=0;
        cost=0;
        while(spfa(s,t,flow,cost));//如果能找到增广路就继续
        return flow;
    }
} mcmf;
int store[55][55],co[55][55][55],od[55][55],cnt[55];
int main()
{
    int n,m,k;
    while(scanf("%d%d%d",&n,&m,&k))
    {
        if(n==0&&n==m&&n==k)
            break;
        memset(cnt,0,sizeof(cnt));
        for(int i=1; i<=n; i++)
        {
            for(int j=1; j<=k; j++)
            {
                scanf("%d",&od[i][j]);
                cnt[j]+=od[i][j];
            }
        }
        for(int i=1; i<=m; i++)
        {
            for(int j=1; j<=k; j++)
            {
                scanf("%d",&store[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",&co[i][j][l]);
                }
            }
        }
        int s=n+m+1,t=n+m+2;
        long long ans=0;
        bool flag=true;
        for(int i=1; i<=k; i++)
        {
            mcmf.init(t);
            for(int j=1; j<=n; j++)
            {
                mcmf.addedge(j,t,od[j][i],0);
            }
            for(int j=1; j<=m; j++)
            {
                mcmf.addedge(s,j+n,store[j][i],0);
            }
            for(int j=1; j<=n; j++)
            {
                for(int l=1; l<=m; l++)
                {
                    mcmf.addedge(l+n,j,inf,co[i][j][l]);
                }
            }
            long long cur;
            if(mcmf.mincostmaxflow(s,t,cur)==cnt[i])
            {
                ans+=cur;
            }
            else
            {
                flag=false;
                break;
            }
        }
        if(flag)
            printf("%lld\n",ans);
        else
            printf("-1\n");
    }
    return 0;
}