题目来源
题目翻译:
解题思路:
本题的解题方法和数组本身没有关系,主要是计算出和3取余等于0,等于1,等于2的个数有多少。在三者不相等的情况下,分情况讨论,找三个数的最大值,让它减,后者加。比如是c0最大,我们就让c0–,c1++(c0-1的话,也就相当于c1+1了,因为c0是和3取余等于0的个数,题目要元素加1,取余为0的+1的话,自然c1就+1了),另外其他的两个就以此讨论就行。
上代码!!
#include<bits/stdc++.h>
using namespace std;
const int MAX_N = 3e4+5;
int a[MAX_N];
int main(){
int t;
cin>>t;
while(t--){
int n;
cin>>n;
int c0=0,c1=0,c2=0;
int ans = 0;
for(int i=0;i<n;i++){
cin>>a[i];
if(a[i]%3==0){
c0++;
}else if(a[i]%3==1){
c1++;
}else c2++;
}
while(c0!=c1||c0!=c2||c1!=c2){
ans++;
if(c0>=c2&&c0>=c1){
c0--;
c1++;
}else if(c1>=c2&&c1>=c0){
c1--;
c2++;
}else{
c2--;
c0++;
}
}
cout<<ans<<endl;
}
return 0;
}