[POJ] 3020 Antenna Placement(二分图最大匹配)

题目地址：http://poj.org/problem?id=3020

输入一个字符矩阵，'*'可行，'o'不可行。因为一个点可以和上下左右四个方向的一个可行点组成一个集合，所以对图进行黑白染色（每个点的值为其横纵坐标之和），然后就可划分为二分图，进行最大匹配。最后最大匹配数加剩下的单个点数量即为所求。

#include<cstdio>
#include<iostream>
#include<string.h>
#include<algorithm>
#include<math.h>
#include<stdbool.h>
#include<time.h>
#include<stdlib.h>
#include<set>
#include<map>
#include<stack>
#include<queue>
#include<vector>
using namespace std;
#define clr(x,y)    memset(x,y,sizeof(x))
#define sqr(x)      ((x)*(x))
#define rep(i,a,b)  for(int i=(a);i<=(b);i++)
#define LL          long long
#define INF         0x3f3f3f3f
#define A           first
#define B           second
#define PI          acos(-1.0)
const int N=2000+131;
const int dx[4]={0,0,-1,1};
const int dy[4]={1,-1,0,0};
int n,m,k,num,k1,k2,f[N],g[N][N],link[N],flag[N][N],b[N],c[N];
char a[N][N];

void init()
{
    clr(f,0);
    clr(g,0);
    clr(link,-1);
    clr(flag,0);
    clr(b,0);
    clr(c,0);
    num=0;
    k1=0;
    k2=0;
}

bool find(int x)
{
    for(int i=0;i<k2;i++) {
        if(!f[c[i]] && g[x][c[i]]) {
            f[c[i]]=1;
            if(link[c[i]]==-1 || find(link[c[i]])) {
                link[c[i]]=x;
                return true;
            }
        }
    }
    
    return false;
}

int hungary()
{
    int ans=0;
    for(int i=0;i<k1;i++) {
        clr(f,0);
        if(find(b[i])) ans++;
    }
    
    return ans;
}

int main()
{
    int u,v,cas;
    
    scanf("%d",&cas);
    while(cas--) {
        init();
        scanf("%d%d",&m,&n);
        for(int i=1;i<=m;i++) {
            scanf("%s",a[i]+1);
            for(int j=1;j<=n;j++) {
                if(a[i][j]=='*') {
                    flag[i][j]=1;
                    num++;
                }
            }
        }
        
        for(int i=1;i<=m;i++) {
            for(int j=1;j<=n;j++){
                int p=(i-1)*n+j;
                if((i+j)&1) {
                    if(flag[i][j]) b[k1++]=p;
                } else {
                    if(flag[i][j]) c[k2++]=p;
                }
            }
        } 
        
        for(int i=1;i<=m;i++) {
            for(int j=1;j<=n;j++) {
                if(flag[i][j]) {
                    for(int k=0;k<4;k++) {
                        int nx=i+dx[k];
                        int ny=j+dy[k];
                        if(flag[nx][ny]) {
                            u=(i-1)*n+j;
                            v=(nx-1)*n+ny;
                            g[u][v]=1;
                        }
                    }
                }
            }
        }
        
        int ans=hungary();
        printf("%d\n",ans+num-2*ans);
        
        
    }
   
    
    
    return 0;
}