bzoj1045 糖果传递

最新推荐文章于 2019-09-30 16:17:50 发布

weixin_34392906

最新推荐文章于 2019-09-30 16:17:50 发布

阅读量67

点赞数

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/huyufeifei/p/9024213.html

本文介绍了一道关于环形纸牌均分的问题，并提供了一个简洁高效的算法解决方案。通过计算前缀和并找到中位数，最终求得最小调整代价。

真·水题。

环形均分纸牌模板题。

只需求出前缀和，然后取中位数求前缀和总和即可。

 1 #include <cstdio>
 2 #include <algorithm>
 3 using namespace std;
 4 typedef long long LL;
 5 const int N = 1000010;
 6 LL a[N], sum[N];
 7 int main() {
 8     int n;
 9     scanf("%d", &n);
10     for(int i = 1; i <= n; i++) {
11         scanf("%d", &a[i]);
12         sum[i] = sum[i - 1] + a[i];
13     }
14     LL k  = sum[n] / n;
15     for(int i = 1; i <= n; i++) {
16         a[i] -= k;
17         sum[i] = sum[i - 1] + a[i];
18     }
19     sort(sum + 1, sum + n + 1);
20     k = sum[(n + 1) >> 1];
21     LL ans = 0;
22     for(int i = 1; i <= n; i++) {
23         ans += abs(sum[i] - k);
24     }
25     printf("%lld\n", ans);
26     return 0;
27 }