codevs 1227 方格取数2

网络流+拆点 π_π（好像HDU上有类似的题。。那时我还是用Pascal写的）

算是裸题吧。。然而我本不会费用流

看了下黄学长的代码，说下自己的理解（黄学长只贴了代码π_π）

把每一个点拆成两个点，一个是流进的，一个是流出的

然后建边。。这两个点之间有这几类边：由流入点指向流出点，流量为1，费用为map[i,j]

                                                                        由流出点指向流入点，流量为0，费用为-map[i,j]（即前一条的反向弧）

                                                                        由流入点指向流出点，流量为k（即取数的次数，也就是最大流量），费用为0

                                                                        （因为共取k次，可能不止一次经过这个点）

                                                                        由流出点指向流入点，流量为0，费用为0（即前一条的反向弧）

蒟蒻表示算不清有多少条边，所以瞥了一眼黄学长的数组范围。。。

然后spfa找最长路，即最大费用。当然也可以建边的时候将费用取相反数，然后用最短路。。。（我感觉这完全是eggpain）

有通路就一直找，找到一条最长路就把答案加上这条路的最大流量乘每条边的费用，直到没有通路为止

一开始WA了，查来查去发现spfa的队列开小了。。。

贴代码吧：

#include<cstdio>
#include<iostream>
#include<cstring>
#include<cmath>
#include<algorithm>
#define maxe 1000010
#define maxn 6010
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
struct re{int from,v,fl,c,next;}ed[maxe];
int e,n,k;
ll ans;
int head[maxn],dui[maxe],dis[maxn],from[maxn];
bool pd[maxn];
inline void ins(int x,int y,int fl,int c){                  //连边，这里正边和反向弧一起连了
	ed[++e].from=x;ed[e].v=y;ed[e].fl=fl;ed[e].c=c;
	ed[e].next=head[x];head[x]=e;
	
	ed[++e].from=y;ed[e].v=x;ed[e].fl=0; ed[e].c=-c;
	ed[e].next=head[y];head[y]=e;
}
void init(){
	scanf("%d%d",&n,&k);
	for(int i=1;i<=n;++i)
	for(int j=1;j<=n;++j){
		int x;
		scanf("%d",&x);
		ins(n*(i-1)+j,n*(i-1)+j+n*n,1,x);
		ins(n*(i-1)+j,n*(i-1)+j+n*n,k,0);
		if(i<n)
		ins(n*(i-1)+j+n*n,n*i+j,k,0);
		if(j<n)
		ins(n*(i-1)+j+n*n,n*(i-1)+j+1,k,0);
	}
	ins(0,1,k,0);
	ins(n*n*2,n*n*2+1,k,0);
}
bool spfa(){                                                //最大费用最大流
	int tou=1,wei=1;
	dui[tou]=0;
	memset(dis,-1,sizeof(dis));
	memset(pd,0,sizeof(pd));
	pd[0]=1;
	dis[0]=0;
	for(;tou<=wei;++tou,pd[dui[tou]]=0){
		int u=dui[tou];
		for(int i=head[u];i;i=ed[i].next){
			int v=ed[i].v;
			if(ed[i].fl>0&&dis[v]<dis[u]+ed[i].c){
				dis[v]=dis[u]+ed[i].c;
				from[v]=i;                 //别忘了记录经过的边，便于算答案
				if(!pd[v])dui[++wei]=v,pd[v]=1;
			}
		}
	}                                                  //以上是找到当前的可行的最大费用
	if(dis[n*n*2+1]==-1)return 0;
	int sum=k;
	for(int i=from[n*n*2+1];i;i=from[ed[i].from])
	sum=min(sum,ed[i].fl);
	for(int i=from[n*n*2+1];i;i=from[ed[i].from]){
		ed[i].fl-=sum;
		ed[i%2?i+1:i-1].fl+=sum;                   //更新反向弧
		ans+=sum*ed[i].c;
	}                                                  //答案加上在最大费用情况下乘最大的流量
	return 1;
}
int main(){
	init();                                            //读入（这次读入优化就不来写了，也就这么几个数。。）
	while(spfa());                                     //因为答案在spfa里面算了，所以主程序一直跑就可以的
	printf("%lld\n",ans);
	return 0;
}

伏地膜黄学长。。