UVALive 5903 Piece it together 二分匹配,拆点 难度:1

https://icpcarchive.ecs.baylor.edu/index.php?option=com_onlinejudge&Itemid=8&page=show_problem&problem=3914

对L研究可以发现

相当于黑-横白,黑-纵白,每个黑白都要被匹配到,其中黑的横纵各两次,

很自然的想到拆点,黑点拆成专门和横白连接的,专门和纵白连接的,

如果2*黑==白数,运行一次二分匹配算法,匹配数==白数则为可行,剩下的都不可行

#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;
const int dx[4]={0,0,1,-1};
const int dy[4]={1,-1,0,0};
char maz[600][601];
int ind[600][600];
int n,m;
bool inmaz(int x,int y){return 0<=x&&x<n&&0<=y&&y<m;}
struct pnt{
        int x,y;
        pnt(){x=y=0;}
        pnt(int x,int y){this->x=x;this->y=y;}
};
pnt pw[600*600],pb[600*600];
int lenw,lenb;
int e[2*600*600][8];
int len[2*600*600];

int mch[2*600*600];
bool used[2*600*600];

bool dfs(int s){
        used[s]=true;
        for(int i=0;i<len[s];i++){
                int t=e[s][i];
                if(mch[t]==-1||(!used[mch[t]]&&dfs(mch[t]))){
                        mch[t]=s;
                        mch[s]=t;
                        return true;
                }
        }
        return false;
}

int match(){
        memset(mch,-1,sizeof(int)*2*lenw);
        int res=0;
        for(int i=0;i<lenw*2;i++){
                if(mch[i]==-1){
                        memset(used,false,2*lenw);
                        if(dfs(i)){
                                res++;
                        }
                }
        }
        return res;
}

int main(){
        int T;
        scanf("%d",&T);
        while(T--){
                scanf("%d%d",&n,&m);
                lenb=lenw=0;
                memset(len,0,sizeof(len));
                memset(ind,0,sizeof(ind));
                for(int i=0;i<n;i++)scanf("%s",maz[i]);
                for(int i=0;i<n;i++){
                        for(int j=0;j<m;j++){
                                if(maz[i][j]=='W'){
                                        ind[i][j]=lenw;
                                        pw[lenw++]=pnt(i,j);
                                }
                                else if(maz[i][j]=='B'){
                                        ind[i][j]=lenb;
                                        pb[lenb++]=pnt(i,j);
                                }
                        }
                }
                if(lenw!=lenb*2){
                        puts("NO");
                        continue;
                }
                for(int i=0;i<lenb;i++){
                        int x=pb[i].x,y=pb[i].y;
                        for(int j=0;j<2;j++){
                                int txj=x+dx[j],tyj=y+dy[j];
                                if(!inmaz(txj,tyj)||maz[txj][tyj]!='W')continue;
                                int indj=ind[txj][tyj],hor=lenw+2*i+1;

                                e[indj][len[indj]++]=hor;
                                e[hor][len[hor]++]=indj;
                        }
                        for(int k=2;k<4;k++){
                                int txk=x+dx[k],tyk=y+dy[k];
                                if(!inmaz(txk,tyk)||maz[txk][tyk]!='W')continue;
                                int indk=ind[txk][tyk],gra=lenw+2*i;
                                e[indk][len[indk]++]=gra;
                                e[gra][len[gra]++]=indk;
                        }
                }
                int res=match();
                if(res!=lenw)puts("NO");
                else puts("YES");
        }
        return 0;
}