HAOI2008 糖果傳遞
阿新 • • 發佈:2018-11-10
一道很有趣的貪心題(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') usingnamespace 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; }