Codeforces Round #702 (Div. 3)B. Balanced Remainders

思路：

答案与数组内元素本身无关，只与其模3后的值的个数有关，于是，可以想到，while(c0!=c1 || c0!=c2 || c1!=c2)，便从c0到c2，找到最大的值，记为pos，然后让其后一个值加上（pos-n/3），直到c0c1c2.

#include<bits/stdc++.h>
using namespace std;
#define ll long long
const int mod = 1e9 + 7;
int c[3];
int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		memset(c,0,sizeof(c));
		int n;
		scanf("%d",&n);
		for(int i = 1; i <= n; i++)
		{
			int a;
			scanf("%d",&a);
			c[a%3]++;
		}
		int pos,k = n/3;
		int ans = 0;
		while(c[0]!=c[1]||c[0]!=c[2]||c[1]!=c[2])
		{
			for(int i = 0; i < 2; i++)
			{
				if(c[i] >= c[i+1])
				{
					pos = i;
					break;
				}
			}
			if(c[2] > c[0] && c[2] > c[1])
			pos = 2;
			ans += (c[pos] - k);
			c[(pos+1)%3] += (c[pos] - k);
		 	c[pos] = k;
		}
		printf("%d\n",ans);
	}
	return 0;
}