HDU 2282 Chocolate（最大权匹配-KM）

Description

有$n$个盒子围成一圈，第$i$个盒子里有$a[i]$个糖，总糖数不超过$n$，一次操作可以把一个盒子里的一个糖放到相邻的盒子里，问至少需要多少次操作可以让每个盒子里至多一个糖

Input

多组用例，每组用例首先输入一整数$n$表示盒子数，之后输入$n$个整数$a[i]$表示第$i$个盒子里糖的数量$(1\le n\le 500)$

Output

输出使得每个盒子里糖至多一个所需的最少操作数

Sample Input

10
1
3
3
0
0
2
0
0
0
0

Sample Output

9

Solution

如果$a[i]>1$，则把第$i$个盒子分成$a[i]-1$个点表示这个盒子里的$a[i]-1$个糖要通过一些操作放到其他盒子里，如果$a[i]=0$，则把第$i$个盒子看作另一排的一个点表示该盒子可以容纳一个糖，对于第一排的点$i$和第二排的点$j$，如果想把$i$所代表第$x$个盒子的一个糖放到$j$所代表的盒子$y$中，则至少需要$min(|y-x|,n-|y-x|)$步操作，以此建边得到一个带权二分图，把边权取负用$KM$算法求出最大二分匹配再取负即为答案

Code

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
using namespace std;
#define maxn 510
#define INF 0x3f3f3f3f
int nx,ny;
int linker[maxn],lx[maxn],ly[maxn],slack[maxn];
int visx[maxn],visy[maxn],w[maxn][maxn];
int dfs(int x)
{
    visx[x]=1;
    for(int y=1;y<=ny;y++)
    {
        if(visy[y])
            continue;
        int tmp=lx[x]+ly[y]-w[x][y];
        if(tmp==0)
        {
            visy[y]=1;
            if(linker[y]==-1||dfs(linker[y]))
            {
                linker[y]=x;
                return 1;
            }
        }
        else if(slack[y]>tmp)
            slack[y]=tmp;
    }
    return 0;
}
int km()
{
    int i,j;
    memset(linker,-1,sizeof(linker));
    memset(ly,0,sizeof(ly));
    for(i=1;i<=nx;i++)  
        for(j=1,lx[i]=-INF;j<=ny;j++)
            if(w[i][j]>lx[i])
                lx[i]=w[i][j];
    for(int x=1;x<=nx;x++)
    {
        for(i=1;i<=ny;i++)
            slack[i]=INF;
        while(1)
        {
            memset(visx,0,sizeof(visx));
            memset(visy,0,sizeof(visy));
            if(dfs(x)) 
                break;  
            int d=INF;
            for(i=1;i<=ny;i++)
                if(!visy[i]&&d>slack[i])
                    d=slack[i];
            for(i=1;i<=nx;i++)
                if(visx[i])
                    lx[i]-=d;
            for(i=1;i<=ny;i++) 
                if(visy[i])
                    ly[i]+=d;
                else
                    slack[i]-=d;
        }
    }
    int res=0;
    for(i=1;i<=ny;i++)
        if(linker[i]!=-1)
            res+=w[linker[i]][i];
    return res;
}
int n,a[maxn],x[maxn],y[maxn];
int main()
{
    while(~scanf("%d",&n))
    {
        nx=ny=0;
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            if(a[i]>1)
            {
                for(int j=1;j<a[i];j++)
                    x[++nx]=i;
            }
            else if(!a[i])y[++ny]=i;
        }
        for(int i=1;i<=nx;i++)
            for(int j=1;j<=ny;j++)
                w[i][j]=-min(abs(y[j]-x[i]),n-abs(y[j]-x[i]));
        printf("%d\n",-km());
    }
    return 0;
}