CODEVS1022 覆盖 (二分图染色+匈牙利算法)

时间:2021-02-10 08:31:17

先对整幅图进行二分图染色,再跑一遍匈牙利算法。

 /* CODEVS1022 */
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <cmath> #define maxn 10008 struct edge{
int u,v,next;
}eg[maxn*]; int dx[]={,,,-};
int dy[]={,-,,};
int a[][];
int cl[maxn];
int n,m,k,sum,ans;
int last[maxn],l[maxn];
bool pd[maxn]; void dfs(int i,int j,int w)
{
int x,y,k;
cl[a[i][j]]=w;
for (k=;k<;k++)
{
int x,y;
x=i+dx[k];
y=j+dy[k];
if ((a[x][y]!=-)&&!cl[a[x][y]])
dfs(x,y,-w);
}
}
void add(int u,int v)
{
eg[++sum].u=u;
eg[sum].v=v;
eg[sum].next=last[u];
last[u]=sum;
}
bool find(int u)
{
for (int i=last[u];i;i=eg[i].next)
{
int v=eg[i].v;
if (!pd[v])
{
pd[v]=;
if ((!l[v])||find(l[v]))
{
l[v]=u;
return ;
}
}
}
return ;
}
int main()
{
int i,j,k;
sum=;
memset(a,-,sizeof(a));
scanf("%d%d%d",&m,&n,&k);
for (i=;i<=m;i++)
for (j=;j<=n;j++)
a[i][j]=(i-)*n+j;
for (int i=;i<=k;i++)
{
int x,y;
scanf("%d",&x);
if (x==) break;
scanf("%d",&y);
a[x][y]=-;
}
for (i=;i<=m;i++)
for (j=;j<=n;j++)
if ((a[i][j]!=-)&&!cl[a[i][j]])
dfs(i,j,);
for (i=;i<=m;i++)
for (j=;j<=n;j++)
if (cl[a[i][j]]) for (k=;k<;k++)
{
int x,y;
x=i+dx[k];
y=j+dy[k];
if ((a[x][y]!=-)&&cl[a[x][y]])
add(a[i][j],a[x][y]);
}
ans=;
memset(l,,sizeof(l));
for (i=;i<=m*n;i++)
if (cl[i]==)
{
memset(pd,,sizeof(pd));
if (find(i)) ans++;
}
//for (i=1;i<=m*n;i++) printf("%d ",l[i]);
printf("%d",ans);
return ;
}