Kaka's Matrix Travels poj 3422 费用流

题目大意

给一个N*N的方阵，从[1,1]到[n,n]走K次，走过每个方格加上上面的数，然后这个格上面的数变为0。求可取得的最大的值。

分析

建图很重要，这里用到拆点，将每个点拆成两个，这两点之间连两条边，一条容量为1，费用为该节点的值，另一条边容量为无穷或k，费用为0。（这样保证就算经过这点k次时，费用也只被计算一次）由于每个点只能往右或者往下走，所以将它和右边及下边的点连一条边，容量为无穷，费用为0.建源点和汇点。

code

#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<string>
#include<algorithm>
#include<queue>;

using namespace std;

struct arr{
    int x,y,w,c,next;
}edge[1000000];
int ls[12000];
int dis[12000];
int f[1000000];
int v[12000];
int sign[12000]; 
int edge_m;
int ans;
int s,t;
int n,m;

int add(int x,int y,int w,int c)
{
    edge_m++;
    edge[edge_m]=(arr){x,y,w,c,ls[x]},f[edge_m]=w,ls[x]=edge_m; 
    edge_m++;
    edge[edge_m]=(arr){y,x,w,-c,ls[y]},f[edge_m]=0,ls[y]=edge_m;
} 

bool bfs()
{
    for (int i=s;i<=t+1;i++) dis[i]=-2000000000;
    memset(v,0,sizeof(v));
    queue<int> q;   
    dis[s]=0;
    v[s]=0;
    q.push(s);
    do
    {
        int x=q.front();
        q.pop();
        for (int i=ls[x];i;i=edge[i].next)
        {
            if ((dis[edge[i].y]<dis[x]+edge[i].c)&&(f[i]))
            {
                dis[edge[i].y]=dis[x]+edge[i].c;
                sign[edge[i].y]=i;
                if (!v[edge[i].y])
                {
                    v[edge[i].y]=1;
                    q.push(edge[i].y);
                }
            }
        }
        v[x]=0;
    }while (!q.empty());
    if (dis[t]!=-2000000000)
        return true;
    else
        return false;
}

void mcf()
{
    int mn=2000000000;
    int x=t;
    while (sign[x])
    {
        mn=min(mn,f[sign[x]]);
        x=edge[sign[x]].x;
    }
    ans+=mn*dis[t];
    x=t;
    while (sign[x])
    {
        f[sign[x]]-=mn;
        f[sign[x]^1]+=mn;
        x=edge[sign[x]].x;
    }
}

int dinic()
{
    while (bfs())
    {
        mcf();
    }
}

int main()
{
    while (scanf("%d%d", &n, &m) != EOF){
    edge_m=1;
    memset(edge,0,sizeof(edge));
    memset(ls,0,sizeof(ls));
    memset(f,0,sizeof(f));
    int a[120][120];
    s=0; t=n*n*2+1;
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++)
            scanf("%d",&a[i][j]);
    for (int i=1;i<=n;i++)
        for (int j=1;j<=n;j++)
        {
            add(n*n+(i-1)*n+j,(i-1)*n+j,m-1,0);
            add(n*n+(i-1)*n+j,(i-1)*n+j,1,a[i][j]);
        }
    add(n*n+1,1,m-1,0);
    add(n*n*2,n*n,m-1,0);
    for (int i=2;i<=n;i++)
        for (int j=2;j<=n;j++)
        {
            add((i-2)*n+j,n*n+(i-1)*n+j,m,0);
            add((i-1)*n+j-1,n*n+(i-1)*n+j,m,0);
        }
    for (int i=2;i<=n;i++)
        add(i-1,n*n+i,m,0);
    for (int i=2;i<=n;i++)
        add((i-2)*n+1,n*n+(i-1)*n+1,m,0);
    add(0,n*n+1,m,0);
    add(n*n,t,m,0);
    ans=0;
    dinic();
    printf("%d\n",ans);
    }
    return 0;
}