pat1003：我要通过！

https://www.patest.cn/contests/pat-b-practise/1002

#include "stdio.h"

int check(char str[])
{
	int t = 0, count1 = 0, count2 = 0, count3 = 0;
	while('\0' != str[t])
	{
		if(('P' == str[t]) || ('A' == str[t]) || ('T' == str[t]));
		else
			break;
		++t;
	}
	for(t = 0; 'A' == str[t]; t++)
		++count1;
	if('P' == str[t++])
	{
		for( ; 'A' == str[t]; t++)
			++count2;
		if('T' == str[t++])
		{
			for( ; ('\0' != str[t]) && ('A' == str[t]); t++)
				++count3;
		}
		else
		{
			printf("NO\n");
			return 0;
		}
	}
	else
	{
		printf("NO\n");
		return 0;
	}
	if((0 == count1) && (0 == count3) && ((1 == count2) || (2 == count2)))
		printf("YES\n");
	else if((0 != count2) && (count1 * count2 == count3))
		printf("YES\n");
	else
		printf("NO\n");

	return 0;
}

int main()
{
	int n, i, j;
	scanf("%d", &n);
	char a[10][100] = {0};
	for(i = 0; i < n; ++i)
		scanf("%s", a[i]);

	for(j = 0; j < n; ++j)
		check(a[j]);

	return 0;
}