CODEVS 1227 方格取数 2

1227 方格取数 2

 时间限制: 1 s

 空间限制: 128000 KB

题目描述 Description

给出一个n*n的矩阵,每一格有一个非负整数Aij,(Aij <= 1000)现在从(1,1)出发,可以往右或者往下走,最后到达(n,n),每达到一格,把该格子的数取出来,该格子的数就变成0,这样一共走K次,现在要求K次所达到的方格的数的和最大

输入描述 Input Description

第一行两个数n,k（1<=n<=50, 0<=k<=10）

接下来n行,每行n个数,分别表示矩阵的每个格子的数

输出描述 Output Description

一个数,为最大和

样例输入 Sample Input

3 1

1 2 3

0 2 1

1 4 2

样例输出 Sample Output

11

数据范围及提示 Data Size & Hint

1<=n<=50, 0<=k<=1

拆点最大费用最大流裸题（本蒟蒻第一次写，打一个模板）

#include<cmath>
#include<ctime>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<iostream>
#include<algorithm>
#include<iomanip>
#include<vector>
#include<string>
#include<queue>
#include<map>
#include<set>
using namespace std;
typedef long long ll;
inline int read()
{
	int x=0,f=1;char ch=getchar();
	while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
	while(ch<='9'&&ch>='0'){x=10*x+ch-'0';ch=getchar();}
	return x*f;
}
const int N=5011;	
int n,m,k,ecnt=1,s=0,t=5010,q[N<<12],dis[N],from[N],last[N];
ll ans;
bool inq[N];
struct EDGE{int fr,to,nt,val,c;}e[N<<10];
inline void add1(int u,int v,int val,int c){e[++ecnt]=(EDGE){u,v,last[u],val,c};last[u]=ecnt;}
inline void add(int u,int v,int val,int c){add1(u,v,val,c);add1(v,u,0,-c);}
bool spfa()
{
	int head=0,tail=1,u;
	q[head]=s;inq[s]=1;
	memset(dis,-1,sizeof(dis));
	dis[s]=0;
	while(head<tail)
	{
		u=q[head++];inq[u]=0;
		for(int i=last[u];i;i=e[i].nt)
		if(dis[e[i].to]<dis[u]+e[i].c&&e[i].val)
		{
			dis[e[i].to]=dis[u]+e[i].c;from[e[i].to]=i;
			if(!inq[e[i].to]){inq[e[i].to]=1;q[tail++]=e[i].to;}
		}
	}
	if(dis[t]==-1)return 0;return 1;
}
void mincf()
{
	int x=0x3f3f3f3f;
	for(int i=from[t];i;i=from[e[i].fr])x=min(x,e[i].val);
	for(int i=from[t];i;i=from[e[i].fr]){ans+=(x*e[i].c);e[i].val-=x;e[i^1].val+=x;}
}
int main()
{
	n=read();k=read();int x;
	for(int i=1;i<=n;i++)
	for(int j=1;j<=n;j++)
	{
		x=read();
		add((i-1)*n+j,(i-1)*n+j+n*n,1,x);
		add((i-1)*n+j,(i-1)*n+j+n*n,k,0);
		if(i<n)add((i-1)*n+j+n*n,i*n+j,k,0);
		if(j<n)add((i-1)*n+j+n*n,(i-1)*n+j+1,k,0);
	}
	add(s,1,k,0);add(n*n*2,t,k,0);
	while(spfa())mincf();
	printf("%lld\n",ans);
	return 0;
}