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

HAOI2008 糖果傳遞

傳送門

一道很有趣的貪心題(bin哥看了1s就會了%%%)

覺得本題似乎是環形均分紙牌,但是其實我們不需要再列舉斷點。首先每個人最後分到的糖果數是固定的,我們設x[i]表示第i個人給了ta左邊的人多少顆糖果(第一個人就給到最後一個人),a[i]表示小朋友原來有多少糖果。那麼就有a[i] - x[i] + x[i+1] = ave,也就得到了x[i+1] = x[i] + ave - a[i].令c[i] = a[i] - ave,也就是x[i+1] = x[i] - c[i]。我們就能把所有的x[i]計算出來。我們的目的是讓這個和最小,所以我們只要把所有的c[i]計算出來然後取中位數即可。

看一下程式碼。

#include<cstdio>
#include<algorithm>
#include<cstring>
#include<iostream>
#include<cmath>
#include<set>
#include<queue>
#define rep(i,a,n) for(int i = a;i <= n;i++)
#define per(i,n,a) for(int i = n;i >= a;i--)
#define enter putchar('\n')

using
namespace std; typedef long long ll; const int M = 1000005; const int INF = 1000000009; const ll mod = 1e9+7; int read() { int ans = 0,op = 1; char ch = getchar(); while(ch < '0' || ch > '9') { if(ch == '-') op = -1; ch = getchar(); } while(ch >= '0' && ch <= '9
') { ans *= 10; ans += ch - '0'; ch = getchar(); } return ans * op; } ll a[M],sum[M],n,ans,ave,cur; int main() { n = read(); rep(i,1,n) a[i] = read(),ave += a[i]; ave /= n; rep(i,1,n) a[i] -= ave,sum[i] = sum[i-1] + a[i]; sort(sum+1,sum+1+n); cur = sum[(n+1) >> 1]; rep(i,1,n) ans += abs(sum[i] - cur); printf("%lld\n",ans); return 0; }