ZOJ 1654 - Place the Robots （二分图最大匹配）

 1 #include<iostream>
 2 #include<cstring>
 3 #include<cmath>
 4 #include<algorithm>
 5 #include<cstdlib>
 6 #include<cstdio>
 7 #include<vector>
 8 using namespace std;
 9 char grid[55][55];
10 bool gl[2505][2505];
11 int gx[55][55],gy[55][55],nx,ny;
12 int m,n;
13 bool vis[2505];
14 int link[2505];
15 void init()
16 {
17     nx=0;
18     ny=0;
19     for(int i=0; i<m; ++i)
20     {
21         bool v=true;
22         for(int j=0; j<n; ++j)
23         {
24             if(grid[i][j]==‘#‘) v=true;
25             else if(grid[i][j]==‘o‘)
26             {
27                 if(v) ++nx;
28                 gx[i][j]=nx;
29                 v=false;
30             }
31         }
32     }
33     for(int j=0; j<n; ++j)
34     {
35         bool v=true;
36         for(int i=0; i<m; ++i)
37         {
38             if(grid[i][j]==‘#‘) v=true;
39             else if(grid[i][j]==‘o‘)
40             {
41                 if(v) ++ny;
42                 gy[i][j]=ny;
43                 v=false;
44             }
45         }
46     }
47     memset(gl,0,sizeof(gl));
48     for(int i=0; i<m; ++i)
49         for(int j=0; j<n; ++j)
50             if(grid[i][j]==‘o‘)
51                 gl[gx[i][j]][gy[i][j]]=true;
52     memset(link,-1,sizeof(link));
53 }
54 bool match(int x)
55 {
56     for(int i=1; i<=ny; ++i)
57         if(gl[x][i]&&!vis[i])
58         {
59             vis[i]=true;
60             if(link[i]==-1||match(link[i]))
61             {
62                 link[i]=x;
63                 return true;
64             }
65         }
66     return false;
67 }
68 int main()
69 {
70     int T,kase=0;;
71     scanf("%d",&T);
72     while(T--)
73     {
74         scanf("%d%d",&m,&n);
75         for(int i=0; i<m; ++i)
76             scanf("%s",grid[i]);
77         init();
78         int ans=0;
79         for(int i=1; i<=nx; ++i)
80         {
81             memset(vis,0,sizeof(vis));
82             if(match(i)) ans++;
83         }
84         printf("Case :%d\n",++kase);
85         printf("%d\n",ans);
86     }
87     return 0;
88 }