Description
有 n n 个盒子围成一圈,第个盒子里有 a[i] a [ i ] 个糖,总糖数不超过 n n ,一次操作可以把一个盒子里的一个糖放到相邻的盒子里,问至少需要多少次操作可以让每个盒子里至多一个糖
Input
多组用例,每组用例首先输入一整数表示盒子数,之后输入 n n 个整数表示第 i i 个盒子里糖的数量
Output
输出使得每个盒子里糖至多一个所需的最少操作数
Sample Input
10
1
3
3
0
0
2
0
0
0
0
Sample Output
9
Solution
如果 a[i]>1 a [ i ] > 1 ,则把第 i i 个盒子分成个点表示这个盒子里的 a[i]−1 a [ i ] − 1 个糖要通过一些操作放到其他盒子里,如果 a[i]=0 a [ i ] = 0 ,则把第 i i 个盒子看作另一排的一个点表示该盒子可以容纳一个糖,对于第一排的点和第二排的点 j j ,如果想把所代表第 x x 个盒子的一个糖放到所代表的盒子 y y 中,则至少需要步操作,以此建边得到一个带权二分图,把边权取负用 KM K M 算法求出最大二分匹配再取负即为答案
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;
}