Day66：HDU5520-Number Link-最小费用最大流

注意一下，代码在C++会TLE，G++可以过

 

AC代码：（在G++上提交）

#include<stdio.h>
#include<iostream>
#include<queue>
#include<string.h>
using namespace std;
#define inf 0x3f3f3f3f

const int N=55*100;
struct node
{
    int to,nextt,cap,flow,cost;
} e[N*N];
bool book[N];
int n,m,tot,s,t,z,pre[10*N],head[10*N],dist[10*N];

void add(int u,int v,int cap,int cost)
{
    e[tot].to=v;
    e[tot].nextt=head[u];
    e[tot].cap=cap;
    e[tot].cost=cost;
    e[tot].flow=0;
    head[u]=tot++;

    e[tot].to=u;
    e[tot].nextt=head[v];
    e[tot].cap=0;
    e[tot].cost=-cost;
    e[tot].flow=0;
    head[v]=tot++;

}

bool SPFA()
{
    for(int i=0; i<=t; i++)
    {
        dist[i]=inf;
        book[i]=0;
        pre[i]=-1;
    }
    book[s]=1;
    dist[s]=0;
    queue<int>Q;
    Q.push(s);
    while(!Q.empty())
    {
        int u=Q.front();
        Q.pop();
        book[u]=0;
        for(int i=head[u]; i!=-1; i=e[i].nextt)
        {
            int v=e[i].to;
            if(e[i].cap>e[i].flow&&dist[v]>dist[u]+e[i].cost)
            {
                dist[v]=dist[u]+e[i].cost;
                pre[v]=i;
                if(book[v]==0)
                {
                    book[v]=1;
                    Q.push(v);
                }
            }
        }
    }
    if(dist[t]!=inf)
        return 1;
    return 0;
}

int MCMF()
{
    int flow=0,cost=0;
    while(SPFA())
    {
        int minn=inf;
        for(int i=pre[t]; i!=-1; i=pre[e[i^1].to])
            minn=min(minn,e[i].cap-e[i].flow);
        for(int i=pre[t]; i!=-1; i=pre[e[i^1].to])
        {
            e[i].flow+=minn;
            e[i^1].flow-=minn;
            cost+=e[i].cost*minn;
        }
        flow+=minn;
    }
    if(z==flow)
        return cost;
    return -1;
}

int main()
{
    int T,pp=1,x,y;
    scanf("%d",&T);
    for(int i=1; i<=T; i++)
    {
        memset(head,-1,sizeof(head));
        scanf("%d %d",&n,&m);
        y=n*m;
        tot=z=0;
        s=n*m*2+1,t=s+1;
        for(int i=1; i<=n; i++)
        {
            for(int j=1; j<=m; j++)
            {
                scanf("%d",&x);
                if(x==0)
                {
                    add(s,(i-1)*m+j,1,0);
                    z++;
                    add((i-1)*m+j+y,t,1,0);
                }
                else
                {
                    if(x%2==1)
                    {
                        add(s,(i-1)*m+j,1,0);
                        z++;
                    }
                    else
                        add((i-1)*m+j+y,t,1,0);
                }
            }
        }
        for(int i=1; i<n; i++)
        {
            for(int j=1; j<=m; j++)
            {
                scanf("%d",&x);
                add((i-1)*m+j,i*m+j+y,1,x);
                add(i*m+j,(i-1)*m+j+y,1,x);
            }
        }
        for(int i=1; i<=n; i++)
        {
            for(int j=1; j<m; j++)
            {
                scanf("%d",&x);
                add((i-1)*m+j,(i-1)*m+j+1+y,1,x);
                add((i-1)*m+j+1,(i-1)*m+j+y,1,x);
            }
        }
        int ans=MCMF();
        printf("Case #%d: %d\n",pp++,ans);
    }
    return 0;
}