【BZOJ】2668: [cqoi2012]交换棋子-最小费用最大流

传送门:bzoj2668

---

题解

可以把交换棋子的过程看作图中某种颜色(黑/白)棋子移动的过程。

一个棋子移动的路径中，除起点终点只交换一次外，其它路径上的点都交换了两次。考虑拆点来满足$i$点最多参与$li{m}_i$次交换。

从这里看到一种只用拆成2个点的做法(好写，常数还小)：
设$i$点代表入度的点为$i$，代表出度的点为$i^{\prime }$。

对于一个在原图和终图中颜色相同的点，必然在图中只作为路径中转点存在，所以流量是平衡的，$i$->$i^{\prime }$连容量大小为$\lfloor \frac{li{m}_i}{2}\rfloor$的边。

而对于一个在原图和终图中颜色不同的点，必然其作为起点/终点多交换了一次，$i$->$i^{\prime }$连容量大小为$\lceil \frac{li{m}_i}{2}\rceil$的边(奇数情况时多余的一次可以留给起点/终点的交换)。

---

代码

#include<bits/stdc++.h>
using namespace std;
const int N=850,M=2e5+10,inf=0x7f7f7f7f;

int n,m,a[22][22],b[22][22],ta,tb,S,T,sum,lim;
int dis[N],vs[N],tim;
int head[N],to[M],nxt[M],w[M],c[M],tot=1,cost,ans;
bool inq[N];
int dx[10]={-1,-1,-1,0,0,1,1,1};
int dy[10]={-1,0,1,-1,1,-1,0,1};
char s[30];

inline int trs(int x,int y,int bs){return (x-1)*m+y+bs*lim;}

inline void lk(int u,int v,int fw,int vv)
{
    to[++tot]=v;nxt[tot]=head[u];head[u]=tot;w[tot]=fw;c[tot]=vv;
    to[++tot]=u;nxt[tot]=head[v];head[v]=tot;w[tot]=0;c[tot]=-vv;
}

deque<int>que;
inline bool bfs()
{
    int i,j,x;
    memset(dis,0x7f,sizeof(int)*(T+3));
    dis[T]=0;que.push_back(T);inq[T]=true;
    for(;que.size();){
        x=que.front();que.pop_front();
        for(i=head[x];i;i=nxt[i]){
            j=to[i];if((!w[i^1]) || (dis[j]<=dis[x]-c[i])) continue;
            dis[j]=dis[x]-c[i];if(inq[j]) continue;
            if(que.empty() || dis[j]<=dis[que.front()]) que.push_front(j);
            else que.push_back(j);inq[j]=true;
        }
        inq[x]=false;
    }
    return (dis[S]<inf);
}

int dfs(int x,int f)
{
    vs[x]=tim;
    if(x==T) return f;
    int i,j,ss=0,res;
    for(i=head[x];i;i=nxt[i]){
        j=to[i];if(vs[j]==tim || (!w[i]) || dis[j]!=dis[x]-c[i]) continue;
        res=dfs(j,min(w[i],f-ss));if(!res) continue;
        ss+=res;cost+=c[i]*res;w[i]-=res;w[i^1]+=res;
        if(ss==f) return ss;
    }
    if(!ss) dis[x]=inf;
    return ss;
}

int main(){
	int i,j,k,x,y,z;
	scanf("%d%d",&n,&m);
	S=n*m*2+1;T=S+1;lim=n*m;
	for(i=1;i<=n;++i){
	   scanf("%s",s+1);
	   for(j=1;j<=m;++j){
	   	  a[i][j]=(s[j]=='1');
	   	  ta+=a[i][j];
	   }
	}
	for(i=1;i<=n;++i){
	   scanf("%s",s+1);
	   for(j=1;j<=m;++j){
	   	  b[i][j]=(s[j]=='1');
	   	  tb+=b[i][j];
	   } 
	}
	if(ta!=tb){printf("-1");return 0;}
	for(i=1;i<=n;++i)
	 for(j=1;j<=m;++j){
	 	if(a[i][j] && (!b[i][j])) lk(S,trs(i,j,0),1,0),sum++;
	 	else if(b[i][j] && (!a[i][j])) lk(trs(i,j,1),T,1,0);
	 }
	int bs=0;
	for(i=1;i<=n;++i){
		scanf("%s",s+1);
	    for(j=1;j<=m;++j){
	  	  bs++;z=s[j]-'0';
	  	  if(a[i][j]!=b[i][j]) lk(bs,bs+lim,(z-1)/2+1,0);
	  	  else lk(bs,bs+lim,z>>1,0);
	      for(k=0;k<8;++k){
	      	  x=i+dx[k];y=j+dy[k];if((x<1) || (y<1) || (x>n) || (y>m)) continue;
	      	  lk(bs+lim,trs(x,y,0),inf,1);
	      }
		}
	}
	for(;bfs();)
		for(vs[T]=tim;vs[T]==tim;){
			tim++;ans+=dfs(S,inf);
		}
   if(ans!=sum) printf("-1");
   else printf("%d",cost);
   return 0;
}