洛谷P2045：方格取数加强版

传送门
一看题意，发现第二次取就是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，费用$A_{i,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;
}