Codevs 1227 方格取数 2(费用流)

1227 方格取数 2

时间限制: 1 s

空间限制: 128000 KB

题目等级 : 大师 Master

查看运行结果

题目描述 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<=10

分类标签 Tags

网络流 费用流 图论

/*
拆点+费用流.
自己yy不对然后看的题解orz.
考虑两个东西.
一是计算答案贡献并保证贡献只计算一次.
二是使图连通.
通过一我们很容易想到拆点后建一条流量为1费用为格子贡献的边.
通过二拆点后建一条流量为k-1费用为0的边.
然后把(i,j)与(i-1,j)和(i,j-1)连接起来.
*/
#include<iostream>
#include<cstdio>
#include<queue>
#define MAXN 10010
#define INF 1e9
using namespace std;
int n,k,ans,cut=1,tot,S,T,head[MAXN],g[51][51],dis[MAXN],b[MAXN],fa[MAXN];
struct data{int u,v,next,c,f;}e[MAXN*2];
queue<int>q;
int read()
{
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9') x=x*10+ch-48,ch=getchar();
return x*f;
}
void add(int u,int v,int c,int f)
{
e[++cut].u=u;e[cut].v=v;e[cut].c=c;e[cut].f=f;e[cut].next=head[u];head[u]=cut;
e[++cut].u=v;e[cut].v=u;e[cut].c=0;e[cut].f=-f;e[cut].next=head[v];head[v]=cut;
}
bool bfs(int t)
{
q.push(S);
for(int i=0;i<=T;i++) dis[i]=-INF;dis[S]=0;
while(!q.empty())
{
int u=q.front();q.pop();b[u]=0;
for(int i=head[u];i;i=e[i].next)
{
int v=e[i].v;
if(dis[v]<dis[u]+e[i].f&&e[i].c)
{
dis[v]=dis[u]+e[i].f,fa[v]=i;
if(b[v]!=t) b[v]=t,q.push(v);
}
}
}
return dis[T]!=-INF;
}
void mincost()
{
int t=1;
while(bfs(t))
{
int tmp=fa[T],x=INF;
while(tmp) x=min(x,e[tmp].c),tmp=fa[e[tmp].u];
tmp=fa[T];
while(tmp)
{
e[tmp].c-=x;
e[tmp^1].c+=x;
tmp=fa[e[tmp].u];
}
ans+=dis[T]*x;
}
return ;
}
int main()
{
n=read(),k=read();S=0,T=2*n*n+1;
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
g[i][j]=read(),tot+=g[i][j];
add(0,1,k,0),add(2*n*n,T,k,0);
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
add((i-1)*n+j,(i-1)*n+j+n*n,1,g[i][j]),
add((i-1)*n+j,(i-1)*n+j+n*n,k-1,0);
for(int i=1;i<=n;i++)
for(int j=1;j<=n;j++)
{
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);
}
mincost();
printf("%d",ans);
return 0;
}