PAT乙级——1003 我要通过！ (20分)（C++）

题目链接：https://pintia.cn/problem-sets/994805260223102976/problems/994805323154440192

“答案正确”是自动判题系统给出的最令人欢喜的回复。本题属于 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

思路：

之前看半天没明白这一题到底是什么意思.......看了各个大佬的blog才明白，给出的字符串首先必须要有PAT，然后给出来的3那一长串的意思是，P之前的A的个数为n1，PT之间的A的个数为n2，T之后的A的个数为n3，要满足n3=n2*n1就可以了。于是写一个评判标准函数，字符串为传入参数，如果遍历字符串既没有P，也没有A，也没有T，那么直接不考虑。然后开始记录P，A，T三个字母的个数，如果P不止一个，或者T不止一个，或者A没有，都算错（A可以有多个）。最后一种情况是通过调用find()函数确定P和A和T的位置，首先P一定要在T之前，其次统计n1,n2,n3，最后看是否满足n3=n1*n2即可。

代码：

#include
#include
#include
using namespace std;
bool judge(string s)///评判函数
{
    int count_p=0,count_a=0,count_t=0;///PAT三种字母的个数
    int pos_p=0,pos_a=0,pos_t=0;///PAT各数字的位置
    int num1=0,num2=0,num3=0;///**1**P**2**T**3**,分别表示位置123的A的个数，满足n3=n1*n2就满足条件
    for(int i=0;i=pos_t)///P必须在T的前面，保证顺序
        return false;
    for(int j=0;jpos_p&&jpos_t&&s[j]=='A')
            num3++;
    }
    if(num3==num1*num2)
        return true;
    else
        return false;
}
int main()
{
    int n;
    string s;
    cin>>n;
    for(int i=0;i>s;
       if(judge(s))
        cout<<"YES"<