codevs 1227 方格取数2

来源:互联网 发布:超级山猫直升机数据 编辑:程序博客网 时间:2024/06/07 23:15

网络流+拆点 π_π(好像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 0x3f3f3f3fusing 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;}


伏地膜黄学长。。

0 0