PAT_1003. 我要通过！(20)

“答案正确”是自动判题系统给出的最令人欢喜的回复。本题属于PAT的“答案正确”大派送 —— 只要读入的字符串满足下列条件，系统就输出“答案正确”，否则输出“答案错误”。

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

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

现在就请你为PAT写一个自动裁判程序，判定哪些字符串是可以获得“ 答案正确 ”的。

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

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

输入样例：

8
PAT
PAAT
AAPATAA
AAPAATAAAA
xPATx
PT
Whatever
APAAATAA

输出样例：

YES
YES
YES
YES
NO
NO
NO
NO

#include<stdio.h>
#define N 1000
int main()
{
	int n,temp,i,j,k=0;
	int flag,flagT,flagP,locationP,locationT;
	char s[N][N];

	scanf("%d",&n);
	temp=n;
	while(n)
	{
		scanf("%s",s[k]);
		k++;
		n--;
	}

	k=0;
    n=temp;

	while(n)
	{
		flag=0,flagP=0,flagT=0;
		for(i=0;s[k][i]!='\0';i++)
		{
			if(s[k][i]!='P'&&s[k][i]!='A'&&s[k][i]!='T')//不能出现PAT之外的字符
			{
				flag=1;  break;
			}
			if(s[k][i]=='P')
			{
				if(flagP!=0) //P只能出现一次
				{
					flag=1;  break;
				}
				if(flagT!=0) //T不能在P之前出现
				{
					flag=1;  break;
				}
				else flagP=1;
			}
			if(s[k][i]=='T')
			{
				if(flagT!=0) //T只能出现一次
				{
					flag=1;  break;
				}
				if(flagP==0) //P要在T之前出现
				{
					flag=1;  break;
				}
				else flagT=1;
			}

		} //for.end

		if(flag==0)
		{
			for(j=0;s[k][j]!='\0';j++) //记录P和T的位置
			{
				if(s[k][j]=='P')
					locationP=j;
				if(s[k][j]=='T')
					locationT=j;
			}
		/*关键点_P和T之间至少一个A，并且P左边A的个数*PT中间A的个数等于T右边A的个数*/
			if(((locationP*(locationT-locationP-1))==(i-locationT-1))&&(locationT-locationP)>1)
			{
				flag=2;
			}
			else flag=1;

		} //if(flag==0).end

		if(flag==2)
			printf("YES\n");
		if(flag==1)
			printf("NO\n");

		n--;
		k++;
	} //while.end 
	
	return 0;
} //main.end