传送门
一看题意,发现第二次取就是0了,那么就想到了网络流(其实是看标签)
费用流建图:
1.首先S向(1,1)连一条费用为0,容量k的边
2.然后(n,n)向T连一条费用为0,容量k的边
3.每个点(i,j)向(i+1,j)和(i,j+1)连一条费用0,容量k的边(此处注意判边界)
4.每个点拆成入点和出点,然后入点向出点连一条容量1,费用
Ai,j
的边
5.然后再入点向出点连一条容量k-1,费用0的边
然后跑最大费用最大流就好了。
(小技巧:其实我们可以将所有费用全部取相反数,然后跑完之后再把答案取相反数就好了)
代码:
#include<cstdio>
#include<cstring>
#include<iostream>
#include<cmath>
#include<algorithm>
#include<cstdlib>
#define ll long long
using namespace std;
inline int read(){
int x=0;char ch=' ';int f=1;
while(ch!='-'&&(ch<'0'||ch>'9'))ch=getchar();
if(ch=='-')f=-1,ch=getchar();
while(ch>='0'&&ch<='9')x=(x<<3)+(x<<1)+(ch^48),ch=getchar();
return x*f;
}
const int N=5e3+10,M=N*30;
int n,k,s,t,tot=-1,all;
int head[N],mark[N],d[N],q[M],cur[N],inq[N],to[M],flow[M],Next[M],cost[M];
inline void addedge(int x,int y,int l,int c){
to[++tot]=y;Next[tot]=head[x];flow[tot]=l;cost[tot]=c;head[x]=tot;
to[++tot]=x;Next[tot]=head[y];flow[tot]=0;cost[tot]=-c;head[y]=tot;
}
inline bool spfa(){
for(int i=s;i<=t;i++)d[i]=0x3f3f3f3f,inq[i]=0;
int l=M/2,r=M/2;q[l]=s;d[s]=0;inq[s]=1;
while(l<=r){
int x=q[l++];
for(int i=head[x];i!=-1;i=Next[i]){
int u=to[i];
if(flow[i]&&d[u]>d[x]+cost[i]){
d[u]=d[x]+cost[i];
if(!inq[u]){
inq[u]=1;
if(d[u]<d[q[l]])q[--l]=u;
else q[++r]=u;
}
}
}
inq[x]=0;
}
return d[t]!=0x3f3f3f3f;
}
int C;
inline int dfs(int x,int a){
mark[x]=1;
if(x==t||!a)return a;
int F=0,f;
for(int &i=cur[x];i!=-1;i=Next[i]){
int u=to[i];
if(!mark[u]&&flow[i]&&d[u]==d[x]+cost[i]&&(f=dfs(u,min(a,flow[i])))>0){
flow[i]-=f;
flow[i^1]+=f;
F+=f;
a-=f;
C+=cost[i]*f;
if(!a)return F;
}
}
return F;
}
inline void MCMF(){
while(spfa()){
mark[t]=1;
while(mark[t]){
for(int i=s;i<=t;i++)mark[i]=0,cur[i]=head[i];
dfs(s,0x3f3f3f3f);
}
}
}
int main(){
memset(head,-1,sizeof(head));
n=read();k=read();all=n*n;s=0;t=all+all+1;
for(int i=1;i<=n;i++){
for(int j=1;j<=n;j++){
int a=read(),id=(i-1)*n+j;
addedge(id,id+all,1,-a);
if(k>1)addedge(id,id+all,k-1,0);
if(j!=n)addedge(id+all,id+1,k,0);
if(i!=n)addedge(id+all,id+n,k,0);
}
}
addedge(s,1,k,0);
addedge(all+all,t,k,0);
MCMF();
printf("%d",-C);
return 0;
}