hdu 5198 简单题

Strange Class

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;

char a[15];
int main()
{
//	freopen("E:\input.txt", "r", stdin);
	int b[3];
	char x, y, z;
	while ((scanf("%s", a)) != EOF)
	{
		memset(b, 0, sizeof(b));
		int len = strlen(a);
		int flag = 1;

		if (len % 3 != 0)  //不是3的倍数
		{
			printf("NO\n");
			continue;
		}
		x = a[0], y = a[len / 3], z = a[len - 1];
		if (x == y || y == z || x == z)   //字符串是同一个字母如：aaaa 
		{
			printf("NO\n");
			continue;
		}
		for (int i = 0; i < len / 3; i++)
		{
			if (a[i] != x || a[len / 3 + i] != y || a[len - 1 - i] != z)   //判断3段是否相等
			{
				printf("NO\n");
				flag = 0;
				break;
			}
		}
		if (flag)
		{
			printf("YES\n");
		}	
	}
	return 0;
}