PAT 1003. 我要通过！(20)

#include <stdio.h>
#include <string.h>
int main()
{
char c[100];
int i,j,n;
int count_P,count_A,count_T,pos_P,pos_T;//分别代表'P','A','T'字符的个数，还有'P','T'的位置
scanf("%d\n",&n);
for(i=0;i<n;i++)
{
gets(c);
count_P = 0;
count_A = 0;
count_T = 0;
pos_P = 0;
pos_T = 0;
for(j=0;j<strlen(c);j++)
{
if(c[j]=='P')
{
count_P++;//记录'P'的个数
pos_P = j;//记录'P'的位置
}
if(c[j]=='A')
count_A++;
if(c[j]=='T')
{
count_T++;
pos_T = j;
}
}
/*
五种情况下是不正确的：
1，
4000
'P','A','T'个数总和大于其长度，说明中间有空格，因为gets()函数是能够将空格读入的
2，'T'在'P'后一位，没有'A'
3，'P'个数不止一个，'T'个数也不止一个
4，当'P'与'T'之间的'A'的个数与'P'之前'A'的个数相乘的个数是要等于'T'后面'A'的个数的，否则为错
*/
if(count_P+count_A+count_T != strlen(c) || pos_T-pos_P<=1 || count_P>1 || count_T>1 || pos_P*(pos_T-pos_P-1)!=strlen(c)-pos_T-1)
printf("NO\n");
else
printf("YES\n");
}
}