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

时间:2023-03-09 04:54:42
[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=+;
const int dx[]={,,-,};
const int dy[]={,-,,};
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,);
clr(g,);
clr(link,-);
clr(flag,);
clr(b,);
clr(c,);
num=;
k1=;
k2=;
} bool find(int x)
{
for(int i=;i<k2;i++) {
if(!f[c[i]] && g[x][c[i]]) {
f[c[i]]=;
if(link[c[i]]==- || find(link[c[i]])) {
link[c[i]]=x;
return true;
}
}
} return false;
} int hungary()
{
int ans=;
for(int i=;i<k1;i++) {
clr(f,);
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=;i<=m;i++) {
scanf("%s",a[i]+);
for(int j=;j<=n;j++) {
if(a[i][j]=='*') {
flag[i][j]=;
num++;
}
}
} for(int i=;i<=m;i++) {
for(int j=;j<=n;j++){
int p=(i-)*n+j;
if((i+j)&) {
if(flag[i][j]) b[k1++]=p;
} else {
if(flag[i][j]) c[k2++]=p;
}
}
} for(int i=;i<=m;i++) {
for(int j=;j<=n;j++) {
if(flag[i][j]) {
for(int k=;k<;k++) {
int nx=i+dx[k];
int ny=j+dy[k];
if(flag[nx][ny]) {
u=(i-)*n+j;
v=(nx-)*n+ny;
g[u][v]=;
}
}
}
}
} int ans=hungary();
printf("%d\n",ans+num-*ans); } return ;
}