Jellyfish and its dream

#include
using namespace std;
const int N = 1e6 + 5;
int a[N], b[N];
int main()
{
	int t;
	cin >> t;
	while (t--)
	{
		int n;
		scanf("%d", &n);
		int cnt1 = 0, cnt2 = 0;
		for (int i = 1; i <= n; i++)
		{
			scanf("%d", &a[i]);			
		}
		b[1] = (a[1] - a[n] + 3) % 3;//这个数组是循环的
		for (int i = 1; i <= n; i++)
		{
			if(i!=1)
				b[i] = (a[i] - a[i - 1] + 3) % 3;//差分数组的计算类似于3进制1-2=2
			if (b[i] == 1)
				cnt1++;
			if (b[i] == 2)
				cnt2++;
		}
		if (cnt1 >= cnt2)
		{
			printf("Yes\n");
		}
		else
		{
			printf("No\n");
		}
	}
	return 0;
}

题目大意：有一个由0-2组成的长度为n的数，如果当前数+1等于下一个数（2+1=0，n+1=1）可以把当前数+1，问能否使所有数都相等

思路：要使所有数都相等就是令差分都为0，如果a1,a2,a3的差分2,1,那么我们可以让a2+1使差分变为0,0如果差分为1,1，我们可以让a2+1使差分变为2,0，如果为0,1，我们可以让a2+1变成1,0所以我们要让差分都变为0，首先我们需要利用0,1变1,0移动1的位置使2都变为0然后继续移动1使1,1，都变为0，因为数组是首尾相接的，所以只要求出原数组差分，统计1的数量如果不少于2就可以