luogu P2045 方格取数加强版

题目传送门：https://www.luogu.org/problemnew/show/P2045

题意：

有一个n*n的矩阵，每一个点都有对应的权值，现在你可以走k次，求你能获得的最大价值。注意：每个点选完之后权值变为0。

思路：

最大费用最大流啊。

1.因为只能选k次，所以开一个超级源点和超级汇点，分别连向(1,1)和(n,n)，流量为k。

2.因为每个点的权值只能选一次，所以拆点。

    [1]每一个点x向x'建一条流量为1，费用为点x的价值的边（因为每一个点的权值只能被选一次）；

    [2]每一个点x向x'建一条流量为无限大，费用为0的边（因为一个点可以走多次）；

3.每一个点向它四周的点连一条流量为无限大，费用为0的边（因为一个点可以走到另一个点走多次）。

最后再跑最大费用最大流即可。

代码：

#include<cstdio>
#include<queue>
#include<cstring>
#include<queue>
#include<algorithm>
#define INF 2147483647
using namespace std;
queue<int> f;
	int n,k,len=-1,st,ed;
	struct node{int x,y,c,d,next;} a[100010];
	int last[5010],dis[5010],pre[5010],pos[5010],p[5010];
	bool bz[5010];
void ins(int x,int y,int c,int d)
{
	a[++len].x=x;a[len].y=y;a[len].c=c;a[len].d=d;a[len].next=last[x];last[x]=len;
}
bool spfa()
{
	memset(bz,true,sizeof(bz));
	bz[st]=false;
	memset(dis,63,sizeof(dis));
	dis[st]=0;
	p[st]=INF;
	f.push(st);
	while(!f.empty())
	{
		int x=f.front();
		bz[x]=true;
		for(int i=last[x];i>=0;i=a[i].next)
		{
			int y=a[i].y;
			if(a[i].c>0&&dis[y]>dis[x]+a[i].d)
			{
				dis[y]=dis[x]+a[i].d;
				pos[y]=x;
				pre[y]=i;
				p[y]=min(p[x],a[i].c);
				if(bz[y])
				{
					f.push(y);
					bz[y]=false;
				}
			}
		}
		f.pop();
	}
	return dis[ed]<1061109567;
}
int flow()
{
	int ans=0;
	while(spfa())
	{
		ans+=p[ed]*dis[ed];
		for(int i=ed;i!=st;i=pos[i])
		{
			a[pre[i]].c-=p[ed];
			a[pre[i]^1].c+=p[ed];
		}
	}
	return ans;
}
int main()
{
	scanf("%d %d",&n,&k);
	int x,size=n*n;
	st=0,ed=2*size+1;
	memset(last,-1,sizeof(last));
	ins(st,1,k,0),ins(1,st,0,0);
	ins(size+size,ed,k,0),ins(ed,size+size,0,0);
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
		{
			int id=(i-1)*n+j;
			scanf("%d",&x);
			ins(id,id+size,1,-x),ins(id+size,id,0,x);
			ins(id,id+size,INF,0),ins(id+size,id,0,0);
			if(j<n) ins(size+id,id+1,INF,0),ins(id+1,size+id,0,0);
			if(i<n) ins(size+id,id+n,INF,0),ins(id+n,size+id,0,0);
		}
	printf("%d",-flow());
}