luogu2045 方格取数加强版

题目

　　https://www.luogu.org/problem/show?pid=2045

题解

　　建图就是每个点拆成两个，假如说一个格点是x，上面是up，左面是left，建边：(x,other(x),1,number)，(other(up),x,inf,0)，(other(left),x,inf,0)，(other(up),other(x),inf,0)，(other(left),other(x),inf,0)，(S,(1,1),1,0)，(S,(1,1),K-1,0)，这样每个格点最多选一次，选完一次之后就可以经过而不选择里面的数字。

　　跑最大费用最大流。

代码

//费用流 
#include <cstdio>
#include <algorithm>
#include <queue>
#include <cstring>
#define inf 0x3f3f3f3f
#define maxn 100000
using namespace std;
int head[maxn], next[maxn], to[maxn], c[maxn], w[maxn], d[6000], pre[maxn], cost, tot=1,
	in[6000], S, T, maxflow, N, K;
queue<int> q;
void adde(int a, int b, int cc, int ww)
{to[++tot]=b;c[tot]=cc;w[tot]=ww;next[tot]=head[a];head[a]=tot;}
void adde2(int a, int b, int cc, int ww){adde(a,b,cc,ww);adde(b,a,0,-ww);}
bool spfa()
{
	int x, p;
	for(int i=1;i<=N*N*2;i++)d[i]=-1;
	in[S]=1;d[S]=0;q.push(S);
	while(!q.empty())
	{
		x=q.front();q.pop();
		for(p=head[x];p;p=next[p])
		{
			if(c[p] and d[to[p]]<d[x]+w[p])
			{
				d[to[p]]=d[x]+w[p];
				pre[to[p]]=p;
				if(!in[to[p]])in[to[p]]=1,q.push(to[p]);
			}
		}
		in[x]=0;
	}
	return d[T]!=-1;
}
void augment()
{
	int f=inf, x;
	for(x=T;x!=S;x=to[pre[x] xor 1])f=min(f,c[pre[x]]);
	for(x=T;x!=S;x=to[pre[x] xor 1])c[pre[x]]-=f,c[pre[x] xor 1]+=f;
	cost+=d[T]*f;
}
int table(int x, int y){return N*(x-1)+y;}
void init()
{
	int i, j, x, y, up, left, number;
	scanf("%d%d",&N,&K);
	for(i=1;i<=N;i++)for(j=1;j<=N;j++)
	{
		scanf("%d",&number);
		x=table(i,j), y=x+N*N, up=table(i-1,j)+N*N, left=table(i,j-1)+N*N;
		if(i!=1)adde2(up,x,inf,0), adde2(up,y,inf,0);
		if(j!=1)adde2(left,x,inf,0),adde2(left,y,inf,0);
		adde2(x,y,1,number);
	}
	S=N*N*2+1, T=table(N,N)+N*N;
	adde2(S,table(1,1),1,0);
	adde2(S,table(1,1)+N*N,K-1,0);
}
int main()
{
	init();
	while(spfa())augment();
	printf("%d",cost);
	return 0;
}