PAT 1003我要通过

PAT 1003我要通过

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

思路：
aPbTc模型，abc只包含字符’A’，所包含的字符’A’个数具备以下关系
 num[c]=num[a]*num[b], num[b]>=1
推导：
 2条件决定了 num[b]=1,num[a]=num[c]
 3条件决定了,若b变成bA即num[b]由1变成2, 则num[c]=num[a]+num[a]，以此类推num[b]=1+n，则num[c]=num[a]+num[a]*n=num[a]*num[b]

/*
 * 计算指定指针之间的字符ch个数，发现其他字符返回0
 * @param head,end指向同一字符串不同位置，head<end
 */
unsigned int getChNum(char* head, char* end, char ch, unsigned int* num)
{
 *num = 0;
 head++;
 while (head != end)
 {
  if (*head == ch)
   *num += 1;
  else
   return 0;
  head++;
 }
 return 1;
}

char str[10][102];
int main(void)
{
 unsigned int num,i;
 char* chp, * cht;//'P'字符指针,'T'字符指针
 unsigned int a, b, c;
 scanf("%d",&num);//输入字符串个数
 for (i = 0; i != num; i++)//依次输入字符串
 {
  scanf("%s", &str[i][1]);
 }
 for (i = 0; i != num; i++)//依次处理字符串
 {
  chp = strstr(&str[i][1], "P"); //查找首个P
  if (chp)
  {
   cht = strstr(chp + 1, "T"); //查找P后面的T
   if (cht)
   {
    if (getChNum(&str[i][0], chp, 'A', &a) &&
     getChNum(chp, cht, 'A', &b) &&
     getChNum(cht,&str[i][strlen(&str[i][1])+1], 'A', &c))
    {
     if ((b >= 1) && (c == a * b))
      printf("YES\n");
     else
      printf("NO\n");
    }
    else
     printf("NO\n");
   }
   else
    printf("NO\n");
  }
  else
   printf("NO\n");
 }
 return 0;
}