P2512 [HAOI2008]糖果传递——解题报告

一、题目链接

P2512 [HAOI2008]糖果传递

二、题目大意

有$n$个小朋友坐成一圈，每人有$a_i$个糖果。每人只能给左右两人传递糖果。每人每次传递一个糖果代价为$1$。求使所有人获得均等糖果的最小代价。

数据范围：$1\le n\le 10^6$。

三、题目分析

1. 假设$T[i]$表示第$i$个人向第$i-1$个人传了$T[i]$张牌 ($T[1]$表示第一个向第$n$个)，值为负表示反向。代价为$\sum 1\times |T[i]|$。

2. 由于传递后每个人的牌都为$average$ ，所以可以列出一系列方程
   $$\begin{array}{c}a[1]+T[2]-T[1]=average\\ a[2]+T[3]-T[2]=average\\ \dots \dots \\ a[n]+T[1]-T[n]=average\end{array}\text{\hspace{1em}(1)}$$

3. 用$T[1]$表示$T[2],T[3],\dots ,T[n]$可得：
   $$\begin{array}{c}T[2]=T[1]-(a[1]-average)\\ T[3]=T[1]-(a[2]+a[1]-2\times average)\\ \dots \dots \\ T[n]=T[1]-(a[n-1]+a[n-2]+\cdots +a[1]-(n-1)\times average)\end{array}\text{\hspace{1em}(2)}$$

4. 设$s_i$为前$i$个人糖果数与平均值的差值的前缀和
   $$\begin{array}{c}T[1]=T[1]-s[n]\\ T[2]=T[1]-s[1]\\ T[3]=T[1]-s[2]\\ ......\\ T[n]=T[1]-s[n-1]\end{array}\text{\hspace{1em}(3)}$$
   操作次数可以表示为$\sum |T[i]|=\sum |T[1]-s[i]|$，那么原问题变成了求$T[1]$的值使得上述求和最小。

5. 此时，对$T[1]$的选择即可转化为数轴上几点$s[i]$到动点$T[1]$距离和的最小值。很显然：

   • 若$n$为奇数，当$T[1]$为$s[i]$中位数时和最小。
   • 若$n$为偶数，当$T[1]$为$s[i]$中间两数之间任何值时和最小。

四、正解程序

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;
typedef long long ll;
ll a[1000010],b[1000010],mid,ans,average,n;
int main()
{
    scanf("%lld",&n);
    for(ll i=1;i<=n;i++)
    {
        scanf("%lld",&a[i]);
        average+=a[i];
    }
    average/=n;
    for(ll i=1;i<=n;i++)
        b[i]=b[i-1]+average-a[i];
    sort(b+1,b+1+n);
    ll temp=b[n/2+1],ans=0;
    for(ll i=1;i<=n;i++)
        ans+=abs(b[i]-temp);
    printf("%lld",ans);
    
    return 0;
}