[PAT B1003]我要通过！

[PAT B1003]我要通过！

题目描述

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

字符串中必须仅有 P、 A、 T这三种字符，不可以包含其它字符；
任意形如 xPATx 的字符串都可以获得“答案正确”，其中 x 或者是空字符串，或者是仅由字母 A 组成的字符串；
如果 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

解析

1. 这道题目是一道数学题，其实并不难，20分的题目我花了好长时间才把它做全对，真的是不知道自己这段时间为什么做题这么低迷
2. 通过审题我们可以发现，首先我们输入的string只能含有P，A，T三个字母，如果含有其他字母立即输出错误，然后我们大致意思就是在b的后面每加一个A字母，那么在c的后面就要加一个a，通过这点我们可以发现判断是不是正确的条件，那就是最左端的A的个数乘以中间的A的个数最终等于右边的A的个数，通过这一点就不难做题了。
3. 我做题的时候疏忽了如果不满足格式的话，pos1和pos2是没有初始化的，所以我把它初始化为-1，代表没找到。所以在做题的时候一定要对某些未初始化的变量提高警惕，有没有可能在运行中忽略了它。

#include <iostream>
#include<string>
using namespace std;
int main() 
{
 int n;
 string str;
 scanf("%d", &n);
 for (int i = 0; i < n; i++) {
  bool judge = true;
  cin >> str;
  int p = 0, a = 0, t = 0; //统计三个字符的个数
  int pos1 = -1, pos2 = -1;
  for (int i = 0; i < str.length(); i++) {
   if (str[i] == 'P') {
    p++;
    pos1 = i;
   }
   else if (str[i] == 'A') a++;
   else if (str[i] == 'T') {
    t++;
    pos2 = i;
   }
   else {
    printf("NO\n");
    judge = false;
    break;
   }
  }
  if (!judge)continue;
  int num1 = pos1, num2 = pos2 - pos1 - 1, num3 = str.length() - pos2 - 1;//分别计算前中后的A的个数
  if ((num2 <= 0 || pos1 < 0 || pos2 < 0 || p>1 || t>1) {
   printf("NO\n");
   continue;
  } //如果出现这种情况直接判断错误
  while (num2 > 0) {
   num3 -= num1;
   num2--;
  }
  if (num3 >= 0) printf("YES\n");
  else printf("NO\n");
 }
 return 0;
}

水平有限，如果代码有任何问题或者有不明白的地方，欢迎在留言区评论；也欢迎各位提出宝贵的意见！