CodeForces 999D Equalize the Remainders
题目
题目大意
给定一个数列 a1,a2,...,aN a 1 , a 2 , . . . , a N ,及一个数 M M ,保证是 N N 的一个因数。
对于每一个,记模 M M 的余数为。记 cr(0≤r≤M−1) c r ( 0 ≤ r ≤ M − 1 ) 为 ai a i 模 M M 的出现次数,现可对每一个元素进行操作,求最少的操作数及最终序列,使得 c1=c2=...=cM−1=NM c 1 = c 2 = . . . = c M − 1 = N M ,若有多解则任意输出一个。
思路
一道非常标准的贪心题。
我们可以先预处理一下:使用vector
记录下
ai
a
i
模
M
M
的所有下标。
使用贪心的思想,当我们找到一个大于的一个值,记为 i i ,则我们向后找一个小于的值,记它为 r r ,尽可能多地从向 r r 移动,即:我们将中的尽可能多的数加上 |i−r| | i − r | 。最后记录下答案,输出即可。
实现细节
注意:数据过大,请使用long long
。
注意:由于余数能够形成一个环,故在找 r r 时,范围是从至 2×M 2 × M 。
提示:我们在删除vector
中的首尾元素时,可以使用成员函数pop_front()
和pop_back()
。其中,pop_front()
时间复杂度为
O(n)
O
(
n
)
,pop_back()
时间复杂度为
O(1)
O
(
1
)
。
正解代码
#include<cstdio>
#include<vector>
#include<algorithm>
using namespace std;
typedef long long ll;
const int Maxn=2*1e5;
ll A[Maxn+5];
int N,M,T;
vector<int> s[Maxn+5];
ll sum;
int main() {
#ifdef LOACL
freopen("in.txt","r",stdin);
freopen("out.txt","w",stdout);
#endif
scanf("%d %d",&N,&M);
for(int i=1;i<=N;i++) {
scanf("%lld",&A[i]);
s[A[i]%M].push_back(i);
}
ll r=0;
T=N/M;
for(int i=0;i<M;i++) {
while(s[i].size()>T) {
r=max(1LL*i,r);
while(s[r%M].size()>=T)r++;
ll tmp=min(s[i].size()-T,T-s[r%M].size());
for(int j=1;j<=tmp;j++) {
ll x=(r%M-i+M)%M;
sum+=x;
A[s[i].back()]+=x;
s[i].pop_back();
s[r%M].push_back(A[s[i].size()]);
}
}
}
printf("%lld\n",sum);
for(int i=1;i<=N;i++)
printf("%lld ",A[i]);
return 0;
}
Thanks For Reading!
如有不妥之处请在评论中指出。