[HAOI2008] 糖果傳遞
阿新 • • 發佈:2017-08-29
全部 scu submit discus bsp sam const turn i+1
1
2
5
4
1045: [HAOI2008] 糖果傳遞
Time Limit: 10 Sec Memory Limit: 162 MB Submit: 4184 Solved: 2026 [Submit][Status][Discuss]Description
有n個小朋友坐成一圈,每人有ai個糖果。每人只能給左右兩人傳遞糖果。每人每次傳遞一個糖果代價為1。
Input
第一行一個正整數n<=987654321,表示小朋友的個數.接下來n行,每行一個整數ai,表示第i個小朋友得到的 糖果的顆數.
Output
求使所有人獲得均等糖果的最小代價。
Sample Input
41
2
5
4
Sample Output
4 設$x[i]$表示第$i$堆從前面一堆拿來的,那麽$a[i]+x[i]-x[i+1]=ave$,然後把$x[2,3,\cdots,n]$全部用$x[1]$表示,可以發現$x[n]=x[1]+sum[n-1]-(n-1)*ave$,然後答案求的是$\sum\left|x[i]\right|$的最小值,求出中位數就行了#include <cstdio> #include <algorithm> using namespace std; typedef long long ll; const int maxn = 1000000 + 10; int a[maxn]; ll b[maxn], ave= 0, sum = 0, ans = 0; int main(){ int n; scanf("%d", &n); for(int i = 1; i <= n; i++){ scanf("%d", a + i); ave += a[i]; } ave /= n; sum = 0; for(int i = 1; i <= n; i++){ b[i] = ave * (i - 1) - sum; sum += a[i]; } sort(b + 1, b + n + 1); int mid = 1 + n >> 1; for(int i = 1; i < mid; i++) ans += b[mid] - b[i]; for(int i = mid + 1; i <= n; i++) ans += b[i] - b[mid]; printf("%lld\n", ans); return 0; }
[HAOI2008] 糖果傳遞