【洛谷P2045】方格取数加强版（费用流）

源点先连(1,1)容量为k，费用为0，表示有k条方案。(n,n)连汇点，同上。

然后向下面，右边的方块连容量为INF，费用为0，仅表示可以联通。

为了满足每个点只能用一次的限制，套路地拆点，并且连流量为1，费用为点权的边。注意有个坑，光是这样一条是不行的，因为我可以不取这个数，因此还要连流量为INF，费用为0的边。

注意这是一个最大费用流，我们一般把所有费用取个负就行了。

#include<bits/stdc++.h>
const int N=55;
const int INF=0x3f3f3f3f;
using namespace std;
int n,k,s,t,val[N][N];
inline int in(int x,int y){return n*(x-1)+y;}
inline int out(int x,int y){return in(x,y)+n*n;}
struct Edge
{
    int to,next,cap,f;
}edge[N*N*N];
int tot=1,first[N*2];
inline void addedge(int u,int v,int w,int f)
{
    tot++;
    edge[tot].to=v; edge[tot].next=first[u]; edge[tot].cap=w; edge[tot].f=f; first[u]=tot;
}
inline void add(int u,int v,int w,int f)
{
	addedge(u,v,w,f); addedge(v,u,0,-f);
} 
int dis[N*2];
bool inque[N*2];
bool spfa()
{
    memset(inque,0,sizeof(inque));
    memset(dis,0x3f,sizeof(dis));
    dis[t]=0; inque[t]=1;
    deque<int> q;
    q.push_back(t);
    while(!q.empty())
    {
        int now=q.front();	q.pop_front(); inque[now]=0;
        for(int u=first[now];u;u=edge[u].next)
        {
            int vis=edge[u].to;
            if(edge[u^1].cap&&dis[now]-edge[u].f<dis[vis])
            {
                dis[vis]=dis[now]-edge[u].f;
                if(!inque[vis])
                {
                    inque[vis]=1;
                    if(!q.empty()&&dis[vis]<dis[q.front()])	q.push_front(vis);	//SLF优化 
                    else q.push_back(vis);
                }
            }
        }
    }
    return dis[s]<INF;
}
bool V[N*2];
int ans=0;
int dfs(int now,int f)
{
    V[now]=1;
    if(now==t)	return f;
    int out=0;
    for(int u=first[now];u;u=edge[u].next)
    {
        int vis=edge[u].to;
        if(!V[vis]&&edge[u].cap&&dis[now]-edge[u].f==dis[vis])
        {
            int w=dfs(vis,min(edge[u].cap,f));
            if(!w)	continue;
            ans+=w*edge[u].f;	edge[u].cap-=w; edge[u^1].cap+=w;	out+=w; f-=w;
            if(!f)	break;
        }
    }
    return out;
}
inline void Maxflow()
{
    int ans=0;
    while(spfa())
    {
        V[t]=1;
        while(V[t])
        {
            memset(V,0,sizeof(V));
            ans+=dfs(s,INF);
        }
    } 
}
int main()
{
	cin>>n>>k;
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
			cin>>val[i][j];
	s=0; t=n*n*2+1;
	add(s,in(1,1),k,0);
	add(out(n,n),t,k,0);
	for(int i=1;i<=n;i++)
		for(int j=1;j<=n;j++)
		{
			add(in(i,j),out(i,j),1,-val[i][j]);
			add(in(i,j),out(i,j),INF,0);
			if(j<n) add(out(i,j),in(i,j+1),INF,0);
			if(i<n)	add(out(i,j),in(i+1,j),INF,0);
		}
	Maxflow();
	cout<<-ans;
	return 0;
}