题目
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;
}