题目
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;
}