洛谷P4003 [国家集训队2017]无限之环 网络流 最小费用最大流

匿名 (未验证) 提交于 2019-12-02 23:47:01

有一个\(n\times m\)棋盘,棋盘上每个格子上有一个水管。水管共有\(16\)种,用一个\(4\)位二进制数来表示当前水管向上、右、下、左有个接口。你可以旋转除了\((0101)_2\)\((1010)_2\)的其他水管,求最少的旋转次数使得水管的每一个接口所在方向都有一个相邻的水管接口与其对应。

这种插头相关的题一般都是网络流,虽然据说插头dp可以过此题。这道题里面要求不存在单独的接口,所有的接口一一对应等价于流量平衡,那么最少的旋转次数就应该想到费用流。

首先把整个棋盘染色,黑点连源点,白格连汇点,这些源汇点的节点流量限制为自己的格子内的接头数。那么如果流量网络满流,那就是此题有解,可以构成流量平衡的情况。

对于每个格子,上下左右拆成四个点,表示这个格子的\(4\)个方向,他们的节点容量为\(1\),用来象征对应接口的流量平衡。

对于一开始不能到达的方向,从初始方向连边至目标方向,费用为最少旋转次数,这样无论怎么旋转,接口对应关系依然存在。

#include<bits/stdc++.h> using namespace std; #define re register int #define in inline  #define ak * #define inf 1e9 #define loc(i,j) ((i-1)*m+j) int col,cnt=1,n,m,s,t,h[10005],dis[10005],l,r,q[10005],pre[10005],vis[10005],mp[10005],ans,maxflow,preflow; struct did{     int u,next,to,f,w; }e[1200005]; char qwq; in int read() {     int yz=0,ioi=1;qwq=getchar();     while(qwq<'0'||qwq>'9') ioi=qwq=='-'?~ioi+1:1,qwq=getchar();     while(qwq>='0'&&qwq<='9') yz=(yz<<3)+(yz<<1)+(qwq^48),qwq=getchar();     return yz ak ioi; } in void add(re a,re b,re c,re d) {     if(!col) swap(a,b);     e[++cnt]=(did){a,h[a],b,c,d},h[a]=cnt;     e[++cnt]=(did){b,h[b],a,0,-d},h[b]=cnt; } int spfa() {     memset(pre,0,sizeof(pre));memset(vis,0,sizeof(vis));     for(re i=s;i<=t;i++) dis[i]=i==s?0:inf;     queue<int>q;q.push(s);vis[s]=1;     while(!q.empty())     {         re i=q.front();vis[i]=0;q.pop();         for(re j=h[i],k;k=e[j].to,j;j=e[j].next)         if(e[j].f&&dis[k]>dis[i]+e[j].w)         {             dis[k]=dis[i]+e[j].w;pre[k]=j;             if(!vis[k]) q.push(k),vis[k]=1;         }     }        return dis[t]<inf; } in void solve() {     re delta=inf;     for(re i=t,p;p=pre[i],i!=s;i=e[p^1].to)     delta=min(delta,e[p].f);     for(re i=t,p;p=pre[i],i!=s;i=e[p^1].to)     e[p].f-=delta,e[p^1].f+=delta;     maxflow+=delta;ans+=delta*dis[t]; } in int f(re i,re j) {return i*(n*m+1)+j;} in int g(re i,re j) {return (i-1)*m+j;} in bool ok(re i,re j) {return (i>=1&&i<=n&&j>=1&&j<=m);} in void turn(re p,re x,re y,re z) {add(f(x,p),f(y,p),1,z);} int main() { //  freopen("infinityloop.in","r",stdin); //  freopen("infinityloop.out","w",stdout);     n=read(),m=read(); s=0;t=4*(n*m+1)+4;     for(re i=1;i<=n;i++) for(re j=1;j<=m;j++) mp[g(i,j)]=read();     for(re i=1;i<=n;i++) for(re j=1;j<=m;j++)     {         re p=g(i,j);col=(i+j)%2;         if(col)         {             for(re k=0;k<=3;k++)              if((mp[p]>>k)&1) add(s,f(k,p),1,0),preflow++;         }         else         {             for(re k=0;k<=3;k++)              if((mp[p]>>k)&1) add(t,f(k,p),1,0);         }         if(col)         {             if(ok(i-1,j)) add(f(0,p),f(2,g(i-1,j)),1,0);             if(ok(i,j+1)) add(f(1,p),f(3,g(i,j+1)),1,0);             if(ok(i+1,j)) add(f(2,p),f(0,g(i+1,j)),1,0);             if(ok(i,j-1)) add(f(3,p),f(1,g(i,j-1)),1,0);         }         switch(mp[p])         {             case 0: break; // 0000             case 5: break; // 0101             case 10: break; // 1010             case 15: break; // 1111             case 1: turn(p,0,1,1);turn(p,0,2,2);turn(p,0,3,1);break; // 0001             case 2: turn(p,1,0,1);turn(p,1,3,2);turn(p,1,2,1);break; // 0010             case 4: turn(p,2,1,1);turn(p,2,0,2);turn(p,2,3,1);break; // 0100             case 8: turn(p,3,0,1);turn(p,3,1,2);turn(p,3,2,1);break; // 1000             case 3: turn(p,0,2,1);turn(p,1,3,1);break; // 0011             case 6: turn(p,1,3,1);turn(p,2,0,1);break; // 0110             case 9: turn(p,0,2,1);turn(p,3,1,1);break; // 1001             case 12: turn(p,2,0,1);turn(p,3,1,1);break; // 1100             case 7: turn(p,0,3,1);turn(p,1,3,2);turn(p,2,3,1);break; // 0111             case 11: turn(p,0,2,2);turn(p,1,2,1);turn(p,3,2,1);break; // 1011             case 13: turn(p,0,1,1);turn(p,2,1,1);turn(p,3,1,2);break; // 1101             case 14: turn(p,3,0,1);turn(p,2,0,2);turn(p,1,0,1);break; // 1110         }     }     while(spfa()) solve();     if(maxflow!=preflow) return puts("-1"),0;     cout<<ans<<endl; } 
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!