http://acm.hdu.edu.cn/showproblem.php?pid=2282
题意:有N个盒子,围成一个圈,有的盒子是空的,有的有一个巧克力,有的有多个巧克力,要求最后每个盒子最多一个巧克力。把巧克力移到相邻的盒子需要一步,求最少的步数。
构图:有两个及以上的盒子裂点,向空盒子连线,再匹配
代码:
#include<iostream>
#include<cstdio>
#include<string>
#include<cstring>
#define inf 999999
#define Min(a,b)a<b?a:b
#define Max(a,b)a>b?a:b
#define nMAX 505
#include<cmath>
using namespace std;
int n,m;
int map[505][505],link[505],lx[505],ly[505],
dn[nMAX],dm[nMAX];
bool vx[505],vy[505];
int abs(int x)
{
if(x<0)return -x;
return x;
}
bool dfs(int x)
{
int j;
vx[x]=1;
for(j=1;j<=m;j++)
{
if(!vy[j]&&lx[x]+ly[j]==map[x][j])
{
vy[j]=1;
if(!link[j]||dfs(link[j]))
{
link[j]=x;
return 1;
}
}
}
return 0;
}
int KM()
{
int i,j,k;
for(i=1;i<=n;i++)
{
lx[i]=-inf;
for(j=1;j<=m;j++)
{
lx[i]=Max(lx[i],map[i][j]);
}
}
memset(ly,0,sizeof(ly));
memset(link,0,sizeof(link));
for(i=1;i<=n;i++)
{
while(1)
{
memset(vx,0,sizeof(vx));
memset(vy,0,sizeof(vy));
if(dfs(i))break;
int MIN=inf;
for(j=1;j<=n;j++)
if(vx[j])
for(k=1;k<=m;k++)
if(!vy[k])
MIN=Min(MIN,lx[j]+ly[k]-map[j][k]);
for(j=1;j<=n;j++)if(vx[j])lx[j]-=MIN;
for(j=1;j<=m;j++)if(vy[j])ly[j]+=MIN;
}
}
int ans=0;
for(j=1;j<=m;j++)
if(link[j])ans+=map[link[j]][j];
return ans;
}
int main()
{
int i,j,k;
while(~scanf("%d",&k))
{
n=0,m=0;
for(i=1;i<=k;i++)
{
scanf("%d",&j);
if(j==0)
dm[++m]=i;
else
{
j-=1;
while(j--)
dn[++n]=i;
}
}
for(i=1;i<=n;i++)
for(j=1;j<=m;j++)
{
map[i][j]=Min(abs(dn[i]-dm[j]),k-abs(dn[i]-dm[j]));
map[i][j]=-map[i][j];
}
int ans=-KM();
printf("%d\n",ans);
}
return 0;
}