HDU 2282 Chocolate KM

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;
}