AtCoder Beginner Contest 313 C
做题链接:AtCoder Beginner Contest 313
问题陈述
给你一个整数序列 A=(A1,A2,…,AN)。你可以执行以下操作任意次数(可能为零)。
- 选择带有 1≤i,j≤N的整数 i和 j。将Ai减少 1,将Aj增加 1。
求使A的最小值与最大值之差最多为 1 所需的最小运算次数。
#### 输入
输入内容由标准输入法提供,格式如下
N
A_1 A_2 ... A_N
#### 输出
将答案打印为整数。
思想:1.给定一个固定的B,求使A等于B所需的最小运算次数
2.在所有最大值和最小值最多相差1的B中,找出一个所需的运算次数最少的,即1
做法:
1.设S是所有k(1<=k<=n)中|ak-bk|的和,在一次操作中, 减少ai会使s改变1,增加aj也会改变1,因此,一次操作最多会使s减少两个,现在,我们的目标是使A等于B,即s=0,因此显然至少需要s/2次操作,并且我们总是可以通过s/2次运算使A=B(证明,如果增加一个数而没有对应的数来减,那就说明A的元素的和>B的元素的和,假设不成立)。因此,这个问题的答案是
2.在所有最大值和最小值最多相差1的B中,找出一个所需运算次数最少的
步骤如下:
首先为了使B的元素之和=A的元素之和,且B的最大值和最小值最多相差1
那么B必须由p的(n-r)份和 (p+1)的r份组成,其中p和r分别是A的元素之和和除以n时的商和余数
题目问题可以简化为:在由p的(n-r)份和(p+)的r份组成的B中,找出一个所有|ak-bk|最小的
直觉上,当i=1,2...n按照Ai的升序排序时,让Bi=p代表i的前(n-r)个实例,让Bi=p+1代表i的后r个实例似乎是最优的。(证明:如果有Ai<Aj&&Bi>Bj),那么交换Bi和Bj不会增加|Ai-Bi|+|Aj-Bj|,这可以简单的按照Ai,Aj,Bi,Bj,的排序来划分情况
另外:时间复杂度为O(nlogn),我们可以选择std::nth_element 在O(n)的时间内解决。
代码:
// Problem: C - Approximate Equalization 2
// Contest: AtCoder - AtCoder Beginner Contest 313
// URL: https://atcoder.jp/contests/abc313/tasks/abc313_c
// Memory Limit: 1024 MB
// Time Limit: 2000 ms
//
// Powered by CP Editor (https://cpeditor.org)#include<bits/stdc++.h>
using namespace std;typedef long long ll;const int N = 2e5+5;int n;
int a[N];int main(){cin>>n;vector<int> a(n);ll sum=0;for(int i=0;i<n;i++){cin>>a[i];sum+=a[i];}sort(a.begin(),a.end());vector<int> b(n,sum/n); //空间为n,初始值为sum/nfor(int i=0;i<sum%n;i++){b[n-1-i]++;} ll ans=0;for(int i=0;i<n;i++){ans+=abs(a[i]-b[i]);}cout<<ans/2<<"\n";return 0; }