LibreOJ #2321.「清华集训 2017」无限之环 费用流

题意

有一个n*m的矩阵，每个格子有些方向上会有接口。设把格子顺时针或逆时针旋转90度的代价为1，且所有格子恰好在相反两端的格子不能旋转，现在要求所有的接口一一配对，问最小代价。
$n*m\le2000$

分析

挺妙的建图。
首先把矩阵黑白染色，然后每个点拆成四个点，分别表示四个位置，源点向所有白点的接口连边，所有黑点的接口向汇点连边。所有相邻格子的对应位置连边。
然后对于每种格子，比如说只有一个接口的格子，就把有接口的位置向所有没有接口的位置连一条费用为1或2的边，取决于需要的操作数。
然后如果最大流等于接口数除以2的话就有解，最小费用流即为答案。
打的话细节不是很多，就是比较烦；然后如果一开始没打对的话调起来会十分心累。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<queue>

const int N=2005;
const int inf=1000000000;

int n,m,s,t,cnt,last[N*4],dis[N*4],ans,flow,pre[N*4],a[N][N],bin[5],sum[20];
bool vis[N*4];
struct edge{int to,next,c,w;}e[N*20];
std::queue<int> que;

int point(int x,int y,int z)
{
    return z*n*m+(x-1)*m+y;
}

void addedge(int u,int v,int c,int w)
{
    e[++cnt].to=v;e[cnt].c=c;e[cnt].w=w;e[cnt].next=last[u];last[u]=cnt;
    e[++cnt].to=u;e[cnt].c=0;e[cnt].w=-w;e[cnt].next=last[v];last[v]=cnt;
}

bool spfa()
{
    for (int i=s;i<=t;i++) dis[i]=inf;
    dis[s]=0;que.push(s);vis[s]=1;
    while (!que.empty())
    {
        int u=que.front();que.pop();
        for (int i=last[u];i;i=e[i].next)
            if (e[i].c&&dis[u]+e[i].w<dis[e[i].to])
            {
                dis[e[i].to]=dis[u]+e[i].w;
                pre[e[i].to]=i;
                if (!vis[e[i].to]) vis[e[i].to]=1,que.push(e[i].to);
            }
        vis[u]=0;
    }
    return dis[t]<inf;
}

void mcf()
{
    flow++;ans+=dis[t];int x=t;
    while (x!=s)
    {
        e[pre[x]].c--;
        e[pre[x]^1].c++;
        x=e[pre[x]^1].to;
    }
}

void build()
{
    s=0;t=n*m*4+1;cnt=1;
    for (int i=1;i<=n;i++)
        for (int j=1;j<=m;j++)
            if (i%2==j%2)
            {
                for (int k=0;k<4;k++)
                    if (a[i][j]&bin[k]) addedge(s,point(i,j,k),1,0);
                if (i-1>=1) addedge(point(i,j,0),point(i-1,j,2),1,0);
                if (j+1<=m) addedge(point(i,j,1),point(i,j+1,3),1,0);
                if (i+1<=n) addedge(point(i,j,2),point(i+1,j,0),1,0);
                if (j-1>=1) addedge(point(i,j,3),point(i,j-1,1),1,0);
                if (sum[a[i][j]]==1)
                {
                    int pos;
                    for (int k=0;k<4;k++) if (a[i][j]&bin[k]) {pos=k;break;}
                    for (int k=0;k<4;k++) if (k!=pos) addedge(point(i,j,pos),point(i,j,k),1,abs(k-pos)==2?2:1);
                }
                else if (sum[a[i][j]]==2)
                {
                    if ((a[i][j]&bin[0])&&(a[i][j]&bin[2])||(a[i][j]&bin[1])&&(a[i][j]&bin[3])) continue;
                    if (a[i][j]&bin[0]) addedge(point(i,j,0),point(i,j,2),1,1);
                    else addedge(point(i,j,2),point(i,j,0),1,1);
                    if (a[i][j]&bin[1]) addedge(point(i,j,1),point(i,j,3),1,1);
                    else addedge(point(i,j,3),point(i,j,1),1,1);
                }
                else if (sum[a[i][j]]==3)
                {
                    int p;
                    for (int k=0;k<4;k++) if (!(a[i][j]&bin[k])) {p=k;break;}
                    for (int k=0;k<4;k++)
                        if (abs(p-k)&1) addedge(point(i,j,k),point(i,j,p),1,1);
                        else if (abs(p-k)==2) addedge(point(i,j,k),point(i,j,p),1,2);
                }
            }
            else
            {
                for (int k=0;k<4;k++)
                    if (a[i][j]&bin[k]) addedge(point(i,j,k),t,1,0);
                if (sum[a[i][j]]==1)
                {
                    int pos;
                    for (int k=0;k<4;k++) if (a[i][j]&bin[k]) {pos=k;break;}
                    for (int k=0;k<4;k++) if (k!=pos) addedge(point(i,j,k),point(i,j,pos),1,abs(k-pos)==2?2:1);
                }
                else if (sum[a[i][j]]==2)
                {
                    if ((a[i][j]&bin[0])&&(a[i][j]&bin[2])||(a[i][j]&bin[1])&&(a[i][j]&bin[3])) continue;
                    if (a[i][j]&bin[0]) addedge(point(i,j,2),point(i,j,0),1,1);
                    else addedge(point(i,j,0),point(i,j,2),1,1);
                    if (a[i][j]&bin[1]) addedge(point(i,j,3),point(i,j,1),1,1);
                    else addedge(point(i,j,1),point(i,j,3),1,1);
                }
                else if (sum[a[i][j]]==3)
                {
                    int p;
                    for (int k=0;k<4;k++) if (!(a[i][j]&bin[k])) {p=k;break;}
                    for (int k=0;k<4;k++)
                        if (abs(p-k)&1) addedge(point(i,j,p),point(i,j,k),1,1);
                        else if (abs(p-k)==2) addedge(point(i,j,p),point(i,j,k),1,2);
                }
            }
}

int main()
{
    scanf("%d%d",&n,&m);
    bin[0]=1;
    for (int i=1;i<=4;i++) bin[i]=bin[i-1]*2;
    for (int i=1;i<bin[4];i++) sum[i]=sum[i>>1]+(i&1);
    int tot=0;
    for (int i=1;i<=n;i++)
        for (int j=1;j<=m;j++)
            scanf("%d",&a[i][j]),tot+=sum[a[i][j]];
    build();
    while (spfa()) mcf();
    if (!(tot&1)&&flow==tot/2) printf("%d",ans);
    else puts("-1");
    return 0;
}