1003 “答案正确”

题目

“ 答案正确 ”是自动判题系统给出的最令人欢喜的回复。本题属于 PAT 的“ 答案正确 ”大派送 ——
只要读入的字符串满足下列条件，系统就输出“ 答案正确 ”，否则输出“ 答案错误 ”。
得到“答案正确”的条件是：
1.字符串中必须仅有 P、 A、 T这三种字符，不可以包含其它字符；
2.任意形如 xPATx 的字符串都可以获得“ 答案正确 ”，其中 x 或者是空字符串，或者是仅由字母 A 组成的字符串；
3.如果 aPbTc 是正确的，那么 aPbATca 也是正确的，其中 a、 b、 c 均或者是空字符串，或者是仅由字母 A 组成的字符串。
现在就请你为 PAT 写一个自动裁判程序，判定哪些字符串是可以获得“ 答案正确 ”的。
输入格式：
每个字符串的检测结果占一行，如果该字符串可以获得“答案正确"，则输出YES,否则输出NO。
输入样例：

8
PAT
PAAT
AAPATAA
AAPAATAAAA
xPATx
PT
Whatever
APAAATAA

输出样例：

YES
YES
YES
YES
NO
NO
NO
NO

重点条件三！

只存在’P’, ‘A’, 'T’三种字符；
‘P’, 'T’只能出现一次并且按照该顺序先后出现；
P&T之间不能没有A；
T之后A的数量 = P之前A的数量 × P&T中间A的数量。

代码

#include<stdio.h>
#include<string.h>
int main()
{   
 int n = 0;
 scanf("%d\n",&n);    
 char c;   
 for(int i = 0;i < n;i++)    
 {       
  int pos = 0,count[3]={0,0,0};   \\使用三个数count[3]记录P之前，P&T之间，T之后A的数量。
  \\为了检测'P', 'T'的出现及次序，使用一个标记变量pos

  while((c=getchar())!='\n')       
 {            
   if(c == 'A')            
  count[pos]++;    \\其值在出现P之前为0（使用count[0]记录P之前的A）       
  else if(c == 'P'&&pos == 0)            
  pos = 1;            \\只有在出现P且其值为0时，将值变为1（使用count[1]记录P&T之间的A）
  else if(c == 'T'&&pos == 1)            
  pos = 2;            \\只有在出现T且其值为1时，将其变为2（使用count[2]记录T之后的A）

  else            
  break;        
  }        
  if(c == '\n'&&pos == 2&&count[1]&&count[2] == count[1]*count[0])        \\这是条件和取
  printf("%s\n","YES");        
  else       
  printf("%s\n","NO");       
   if(c !='\n')        
   while(getchar()!='\n');           
    }    
    return 0;
}