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

题目

原题地址

题意

令一个长度为n的数组中，存在n/3个能够被3整除的的数，n/3个除以3余1的数，n/3个除以3余2的数，且对于每个a[i]每次只能进行+1处理。问最少操作多少次；

思路

定义c0，c1，c2三个累加器对a[i]中符合余数分别为0，1，2的a[i]的个数进行累加；
因为c0，c1，c2三个数最后一定相等且等于n/3，故题意可简化为令c0，c1，c2相等；而且如若c0，c1，c2三个数不相等，那么只有（ c0 >= c1 && c0 >= c2 ）；（ c1 >= c0 && c1 >= c2 ）；（ c2 >= c0 && c2 >= c1 )；
所以
c0 - - => c1 ++ ;
c1 - - => c2 ++ ;
c2 - - => c0 ++ ;

AC代码

#include<iostream>
using namespace std ; 

int a[30010] ;  
int main()
{
	int t ; 
	cin >> t ; 
	
	while( t -- ) 
	{
		int n , c0 = 0 , c1 = 0 , c2 = 0 , sum = 0 ; 
		cin >> n ; 
		
		for( int i = 0 ; i < n ; i ++ ) 
		{
			cin >> a[i] ; 
			if( a[i] % 3 == 0 ) c0 ++ ; 
			if( a[i] % 3 == 1 ) c1 ++ ; 
			if( a[i] % 3 == 2 ) c2 ++ ; 
		}
		
		while( c0 != c1 || c0 != c2 || c2 != c1 ) 
		{
			sum ++ ; 
			if( c0 >= c1 && c0 >= c2 ) c0 -- , c1 ++ ; 
			else if( c1 >= c0 && c1 >= c2 ) c1 -- , c2 ++ ;
			else if( c2 >= c0 && c2 >= c1 ) c2 -- , c0 ++ ; 
		} 
		cout << sum << endl ; 
	}
	return 0 ; 
}