HAOI 2008 糖果传递 题解

题目传送门

题目大意： $n$ 个人成环的均分纸牌。

题解

设 $b[i]$ 表示第 $i$ 个人向第 $i+1$ 个人传递的糖果数量，$c[i]$ 表示每个人的初始糖果数量，$ave$ 表示 $c[i]$ 的平均数，也就是最后每个人拥有的糖果的数量，那么可以得到这么一条柿子：

$c[i]+b[i-1]-b[i]=ave$原来有的+前一个人给的-自己给后一个人的=最后的=ave

注意，$b[i]$ 的值可以是负数，表示从后一个人拿 $-b[i]$ 个

然后整理一下

$$\begin{array}{rl}b[i]-b[i-1]& =c[i]-ave\\ b[i+1]-b[i]& =c[i+1]-ave\end{array}$$

那么发现，其实 $b[i]$ 之间两两的差是一定的，那么设 $a[i]=b[i+1]-b[i]$。

于是可以得到下面这堆式子：

$b[1]=b[1]$
$b[2]=b[1]+a[1]$
$b[3]=b[1]+a[1]+a[2]$
$...$
$b[n]=b[1]+a[1]+a[2]+...+a[n-1]$

$\because ans={\sum }_{i=1}^n|b[i]|$
$\therefore ans={\sum }_{i=1}^n|b[1]+{\sum }_{j=1}^{i-1}a[j]|$

设 $d[i]={\sum }_{j=1}^{i}a[j]$

那么有：
$$\begin{array}{rl}ans& =\sum _{i=1}^n|b[1]+d[i-1]|\\ & =\sum _{i=1}^n|b[1]-(-d[i-1])|\end{array}$$

因为所有的 $d[i]$ 都是可以求出来的，那么现在问题就变成了，$b[1]$ 等于多少时，$ans$ 有最小值。

显然地，当 $b[1]$ 等于所有 $-d[i]$ 的中位数时，$ans$ 有最小值。

代码如下：

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
#define ll long long

typedef ll kk[1000010];
kk a,c,d;
ll ave=0;
int n;
inline char cn()
{
    static char buf[1000010],*p1=buf,*p2=buf;
    return p1==p2&&(p2=(p1=buf)+fread(buf,1,1000000,stdin),p1==p2)?EOF:*p1++;
}
void read(ll &x)
{
    x=0;char ch=cn();
    while(ch<'0'||ch>'9')ch=cn();
    while(ch>='0'&&ch<='9')x=x*10+(ch-'0'),ch=cn();
}
void readint(int &x)
{
    x=0;char ch=cn();
    while(ch<'0'||ch>'9')ch=cn();
    while(ch>='0'&&ch<='9')x=x*10+(ch-'0'),ch=cn();
}

int main()
{
	readint(n);
	for(int i=1;i<=n;i++)
	read(c[i]),ave+=c[i];
	ave/=(ll)n;
	for(int i=1;i<n;i++)
	d[i]=d[i-1]-c[i+1]+ave;
	sort(d,d+n);
	ll zws,ans=0;
	if(n%2==1)zws=d[n/2];
	else zws=(d[n/2-1]+d[n/2])/2;
	for(int i=0;i<n;i++)
	ans+=abs(zws-d[i]);
	printf("%lld",ans);
}