PTA 1003 我要通过

得到“答案正确”的条件是：

1. 字符串中必须仅有 P、 A、 T这三种字符，不可以包含其它字符；
2. 任意形如 xPATx 的字符串都可以获得“答案正确”，其中 x或者是空字符串，或者是仅由字母 A 组成的字符串；
3. 如果 aPbTc 是正确的，那么 aPbATca 也是正确的，其中 a、 b、 c均或者是空字符串，或者是仅由字母 A 组成的字符串。

输入格式：

每个测试输入包含 1 个测试用例。第 1 行给出一个正整数 n (≤10)，是需要检测的字符串个数。接下来每个字符串占一行，字符串长度不超过 100，且不包含空格。

输出格式：

每个字符串的检测结果占一行，如果该字符串可以获得“答案正确”，则输出 YES，否则输出 NO。

#include <stdio.h> 
int main() 
{
	int pass,i,n;
	char ch;
	scanf("%d",&n);
	getchar();
	while(n--)
	{
		int temp[3] ={0};
		pass = 1; i=0;
		while( (ch=getchar()) != '\n')
		{
			if(ch == 'A')
				temp[i] ++;
			else if(ch == 'P')
				i=1;
			else if(ch =='T')
				i=2;
			else 
				pass =0;
		}
		if( pass!=0 && temp[1]!=0 && i==2 && temp[0] * temp[1] ==temp[2])
			printf("YES\n");
		else
			printf("NO\n");
	}
    return 0;
}