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;

}