poj3422 Kaka‘s Matrix Travels(k次方格取数 最大费用最大流)

题目

n*n(n<=50)的矩阵a[]，a[i][j]表示(i,j)的苹果数(a[i][j]<=1e3)，

每次你可以从a[1][1]出发，只能向右或向下走一格，

获得这个格子的苹果，苹果被取走后会消失，

你可以走k(0<=k<=10)次，问最多能取得多少苹果

思路来源

栗子巨巨

题解

如果k<=2，这是很经典的dp问题，方格取数，但k一大就不好做了，

由于是点有权值，很自然地想到拆点，

把每个点拆成一个入点和一个出点，点权转化成拆的两个点之间的边权，

作最大费用最大流，第一次经过这条边权值为苹果数，以后的权值就为0，

所以就考虑建两条边，一条流量为1，权值为苹果数；另一条流量为k-1，权值为0，

格子与格子之间，就出点指向下一个格子的入点，流量为k，权值为0

板子是最小费用最大流的板子，取反即可，神奇地统一了k=0的情况

代码

//Copyright(c)2018 Mstdream
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
using namespace std;
const int N=55*55*2,P=55;
const int M=8*N;
int n,k,a[P][P];
int nxt[M],sz=1,to[M],cost[M],flow[M],exsit[N];
int fir[N],q[N*N+10],pre[N],dis[N];
int l[N],r[N],w[N],x[N],c;
void init(){
    memset(fir,0,sizeof fir);
    memset(exsit,0,sizeof exsit);
    sz=1;
    c=0;
}
void add1(int x,int y,int z,int w){
    nxt[++sz]=fir[x];fir[x]=sz;
    to[sz]=y;flow[sz]=z;cost[sz]=w;
}
void add(int x,int y,int z,int w){//无向图需正反调用两次add
    add1(x,y,z,w),add1(y,x,0,-w);
    //fprintf(stderr,"%d %d %d %d\n",x,y,z,w);
}
bool spfa(int S,int T){
    memset(dis,63,sizeof dis);
    int s=0,t=1;q[1]=S;dis[S]=0;
    while(s^t){
        int v=q[++s];exsit[v]=0;
        for(int u=fir[v];u;u=nxt[u]){
            if(flow[u]&&dis[to[u]]>dis[v]+cost[u]){
                dis[to[u]]=dis[v]+cost[u];pre[to[u]]=u;
                if(!exsit[to[u]]){
                    exsit[to[u]]=1;q[++t]=to[u];
                }
            }
        }
    }
    return dis[T]<=1000000000;//最短路的INF
}
int end(int S,int T){
    int now=T,mi=1000000000;
    while(now^S){
        mi=min(mi,flow[pre[now]]);
        now=to[pre[now]^1];
    }
    now=T;int ret=0;
    while(now^S){
        ret+=mi*cost[pre[now]];//ans+=mi ans为流量
        flow[pre[now]]-=mi;
        flow[pre[now]^1]+=mi;
        now=to[pre[now]^1];
    }
    return ret;
}
int mcmf(int S,int T){
    int ret=0;
    while(spfa(S,T)){
        ret+=end(S,T);
    }
    return ret;
}
bool ok(int x,int y){
    return x>=0 && x<n && y>=0 && y<n;
}
int f(int i,int j,int op){
    return op*n*n+i*n+j;
}
int main(){
    while(~scanf("%d%d",&n,&k)){
        init();
        for(int i=0;i<n;++i){
            for(int j=0;j<n;++j){
                scanf("%d",&a[i][j]);
            }
        }
        for(int i=0;i<n;++i){
            for(int j=0;j<n;++j){
                add(f(i,j,0),f(i,j,1),1,-a[i][j]);
                add(f(i,j,0),f(i,j,1),k-1,0);
                if(ok(i+1,j)){
                    add(f(i,j,1),f(i+1,j,0),k,0);
                }
                if(ok(i,j+1)){
                    add(f(i,j,1),f(i,j+1,0),k,0);
                }
            }
        }
        printf("%d\n",-mcmf(f(0,0,0),f(n-1,n-1,1)));
    }
    return 0;
}