【LGP2045】方格取数加强版

题目

还纠结了一下是费用流还是最小割

最终还是决定让最小割去死吧

我们的问题就是让一个点的点权只被计算一次

考虑拆点

1. 将所有点拆成入点和出点，入点向出点连流量为(1)的边

2. 每一个出点往下连能到达的点，向入点连费用为该点点权容量为(0)的边，向出点连费用为(0)容量为(k-1)的边

这样我们就能保证一个点的点权只被计算一次了

代码

#include<queue>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
#define re register
#define LL long long
#define max(a,b) ((a)>(b)?(a):(b))
#define min(a,b) ((a)<(b)?(a):(b))
const int maxn=5005;
const int inf=99999999;
inline int read() {
	char c=getchar();int x=0;while(c<'0'||c>'9') c=getchar();
	while(c>='0'&&c<='9') x=(x<<3)+(x<<1)+c-48,c=getchar();return x;
}
struct E{int v,nxt,w,f;}e[maxn*100];
const int dx[]={0,1};
const int dy[]={1,0};
std::queue<int> q;
int n,num=1,S,T,m,a[51][51],out[51][51],in[51][51];
int head[maxn],vis[maxn],d[maxn];
inline void C(int x,int y,int w,int f) {
	e[++num].v=y;e[num].nxt=head[x];head[x]=num;
	e[num].w=w;e[num].f=f;
}
inline void add(int x,int y,int w,int f) {C(x,y,-1*w,f),C(y,x,w,0);}
inline int SPFA() {
	for(re int i=S;i<=T;i++) vis[i]=0,d[i]=inf;
	d[T]=0,q.push(T);
	while(!q.empty()) {
		int k=q.front();q.pop();vis[k]=0;
		for(re int i=head[k];i;i=e[i].nxt)
		if(e[i^1].f&&d[e[i].v]>d[k]+e[i^1].w) {
			d[e[i].v]=d[k]+e[i^1].w;
			if(!vis[e[i].v]) q.push(e[i].v),vis[e[i].v]=0;
		}
	}
	return d[S]<inf;
}
int dfs(int x,int now) {
	if(x==T||!now) return now;
	int flow=0,ff;vis[x]=1;
	for(re int i=head[x];i;i=e[i].nxt)
	if(e[i].f&&!vis[e[i].v]&&d[e[i].v]==d[x]+e[i^1].w) {
		ff=dfs(e[i].v,min(now,e[i].f));
		if(ff<=0) continue;
		flow+=ff,now-=ff,e[i].f-=ff,e[i^1].f+=ff;
		if(!now) break;
	}
	return flow;
}
int main() {
	n=read(),m=read();
	for(re int i=1;i<=n;i++)
		for(re int j=1;j<=n;j++) a[i][j]=read();
	for(re int i=1;i<=n;i++)
		for(re int j=1;j<=n;j++) in[i][j]=++T,out[i][j]=++T;
	++T;
	for(re int i=1;i<=n;i++)
		for(re int j=1;j<=n;j++) add(in[i][j],out[i][j],0,1);
	for(re int i=1;i<=n;i++)
		for(re int j=1;j<=n;j++) 
			for(re int k=0;k<2;k++)	{
				int x=i+dx[k],y=j+dy[k];
				if(x<1||y<1||x>n||y>n) continue;
				add(out[i][j],in[x][y],a[x][y],1);
				add(out[i][j],out[x][y],0,inf);
			}
	int ans=0;
	add(S,in[1][1],a[1][1],1);add(S,out[1][1],0,m-1);
	add(out[n][n],T,0,m);
	while(SPFA()) {
		vis[T]=1;
		while(vis[T]) {
			for(re int i=S;i<=T;i++) vis[i]=0;
			ans-=dfs(S,inf)*d[S];
		}
	}
	printf("%d
",ans);
	return 0;
}