1. 程式人生 > >[HAOI2008] 糖果傳遞

[HAOI2008] 糖果傳遞

全部 scu submit discus bsp sam const turn i+1

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

4
1
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] 糖果傳遞