首页 > 代码库 > UVA 1514 Piece it together (二分图匹配)

UVA 1514 Piece it together (二分图匹配)

 

【题目链接】 Link

 

【题目大意】

  给你一些由一块黑块和两块白块组成的L形拼图,问你是否能够拼成给出的图

 

【题解】

  我们将所有的黑块拆点,拆分为纵向和横向,和周围的白块连边,
  如果能够得到完美匹配,则说明可以拼成,否则就不能

 

【代码】

#include <cstdio>#include <algorithm>#include <cstring>#include <vector> using namespace std;const int MAX_V=500000;int V,match[MAX_V];vector<int> G[MAX_V];bool used[MAX_V];void add_edge(int u,int v){    G[u].push_back(v);    G[v].push_back(u);}bool dfs(int v){    used[v]=1;    for(int i=0;i<G[v].size();i++){        int u=G[v][i],w=match[u];        if(w<0||!used[w]&&dfs(w)){            match[v]=u;            match[u]=v;            return 1;        }    }return 0;}int bipartite_matching(){    int res=0;    memset(match,-1,sizeof(match));    for(int v=0;v<V;v++){        if(match[v]<0){            memset(used,0,sizeof(used));            if(dfs(v))res++;        }    }return res;}void clear(){for(int i=0;i<V;i++)G[i].clear();}const int MAX_N=500;char mp[MAX_N][MAX_N];int N,M;const int dx[]={1,-1,0,0};const int dy[]={0,0,1,-1};int g[MAX_N][MAX_N];bool check(int x,int y){return(x<0||x>=N||y<0||y>=M||mp[x][y]!=‘W‘);}void solve(){    scanf("%d%d",&N,&M);    memset(g,0,sizeof(g));    int B=0,W=0;    for(int i=0;i<N;i++)scanf("%s",mp[i]);    for(int i=0;i<N;i++){           for(int j=0;j<M;j++){              if(mp[i][j]==‘B‘)g[i][j]=B++;              else if(mp[i][j]==‘W‘)g[i][j]=W++;         }    }if(W!=B<<1){puts("NO");return;}    V=W<<1;clear();    for(int i=0;i<N;i++){        for(int j=0;j<M;j++){            if(mp[i][j]==‘B‘){                for(int k=0;k<4;k++){                    int x=i+dx[k],y=j+dy[k];                    if(check(x,y))continue;                    if(dx[k])add_edge(g[i][j]+B,g[x][y]+B*2);                    else add_edge(g[i][j],g[x][y]+B*2);                }            }        }    }if(bipartite_matching()==W)puts("YES");      else puts("NO"); }int T;int main(){    scanf("%d",&T);    while(T--)solve();    return 0;}

UVA 1514 Piece it together (二分图匹配)