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

伏地膜黄学长。。