Description
有n个小朋友坐成一圈,每人有ai个糖果。每人只能给左右两人传递糖果。每人每次传递一个糖果代价为1。
Input
小朋友个数n 下面n行 ai
Output
求使所有人获得均等糖果的最小代价。
Sample Input
1
2
5
4
Sample Output
HINT
数据规模
30% n<=1000
100% n<=100000
题解:
链状均分纸牌的升级版(环状);
先说说链状:
p[i]表示要从下一堆中要拿的糖果,a[i]表示第i个小朋友拥有的糖,k表示每个小朋友最后应该拥有的糖果数量。
则p[i]=k-a[i]+p[i-1](可以想想看,一下就可以想出来的)
所以最后答案是p[1]+p[2]+...+p[n]。
好了,现在说环状怎么解决:
目的就在于选那个点为第一个点,对p数列取中位数,就可以了。
记sum[i]=sigma(p[j]-p[j-1]) j<=i 那么可以得到p[1]=p[i]-sum[i] 即 p[i]=sum[i]+p[1]
又因为最后的答案ans=sigma p[i]=sigma (s[i]+p[1]) 即数轴上s数组的点到-p[1]的距离之和、
而s数组中的值的集合在取任一点为起点的时候都是不变的、
所以当p[1]为该数组中的中位数时ans最小、
code:
#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define inf 2000000000
#define ll long long
using namespace std;
inline int read()
{
int x=0,f=1;char ch=getchar();
while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
return x*f;
}
int n;
int a[100005],b[100005];
int ans1,ans2;
set<int> q;
int solve(int a[],int b[])
{
int l1=1,r1=n,l2=1,r2=n,ans=0;
while(l1<=r1&&l2<=r2)
{
if(a[l1]>b[l2]){ans+=2;l1++;l2++;}
else if(a[r1]>b[r2]){ans+=2;r1--;r2--;}
else {ans+=(a[l1]==b[r2]);l1++;r2--;}
}
return ans;
}
int main()
{
n=read();
for(int i=1;i<=n;i++)a[i]=read();
for(int i=1;i<=n;i++)b[i]=read();
sort(a+1,a+n+1);
sort(b+1,b+n+1);
printf("%d %d\n",solve(a,b),2*n-solve(b,a));
return 0;
}