bzoj1045题解

【解题思路】

　　（数据范围劝退?正确范围应该是n≤1000000）

　　设x_i表示第i个小朋友给第i+1个小朋友的糖果数（特殊的，x_n表示第n个小朋友给第1个小朋友的糖果数），Â表示平均糖果数，有如下方程组：

•a_i-x_i+x_i-1=Â(i∈[2,n]∩N)

•a₁-x₁+x_n=Â

　　于是可整理为：

•x₂=a₁+x₁-Â

•x₃=a₂+x₂-Â=a₁+a₂+x1-2Â

...

•x_n=a_n-1+x_n-1-Â=...=a₁+a₂+...+a_n-1+x₁-(n-1)Â

　　设s_n=nÂ-∑a_i(i∈[1,n]∩N)，可整理为：

•x₂=x₁-s₁

•x₃=x₁-s₂

...

•x_n=x₁-s_n-1

　　而答案表达式为∑|x_i|(i∈[1,n]∩N)，展开即|x₁|+∑|x₁-s_i-1|(i∈[2,n]∩N)，显然达到最小值时x₁值为所有s_i及0的中位数。复杂度O(nlog₂n)。

【参考代码】

 #pragma GCC optimize(2)
 #include <algorithm>
 #include <cstdio>
 #define REP(i,low,high) for(register int i=(low);i<=(high);i++)
 using namespace std;
 static int n; long long a[],s[];
 int main()
 {
     scanf("%d",&n);
     long long ave=0ll; REP(i,,n) scanf("%lld",a+i),ave+=a[i];
     ave/=n,s[n]=0ll; REP(i,,n-) s[i]=s[i-]+a[i]-ave; sort(s+,s+n+);
     long long m=n&?s[n+>>]:s[n>>]+s[n+>>]>>,ans=0ll;
     REP(i,,n) ans+=llabs(m-s[i]); return printf("%lld\n",ans),;
 }