Balanced Remainders

Problem - 1490B - Codeforces

思路：因为每次都只能将某个数加一，那么实际上C0,C1,C2的数目变化是由规律的，C1只能通过C0中的数变化得到，依次类推，当三个数都相同时，就是答案了

以下是代码实现：

#include<iostream>
#include<vector>
using namespace std;

void solved()
{
    int n;
    cin>>n;
    vector<int>vt;
    int cnt[3]={0};
    for(int i=1;i<=n;i++)
    {
        int num;
        cin>>num;
        vt.push_back(num);
        cnt[num%3]++;
    }
    int ans=0;
    while(!(cnt[0]==cnt[1]&&cnt[1]==n/3))
    {
        for(int i=0;i<3;i++)
        {
            if(cnt[i]>n/3)
            {
                ans+=cnt[i]-n/3;
                if(i==2) cnt[0]+=cnt[i]-n/3;
                else cnt[i+1]+=cnt[i]-n/3;
                cnt[i]=n/3;
            }
        }
    }
    cout<<ans<<endl;
}
int main()
{
    ios::sync_with_stdio(false);
    cin.tie(0);
    cout.tie(0);
    int t;
    cin>>t;
    while(t--)
    {
        solved();
    }
    return 0;
}