pta 1003 我要通过

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

得到“答案正确”的条件是：

字符串中必须仅有 P、 A、 T这三种字符，不可以包含其它字符；
任意形如 xPATx 的字符串都可以获得“答案正确”，其中 x 或者是空字符串，或者是仅由字母 A 组成的字符串；
如果 aPbTc 是正确的，那么 aPbATca 也是正确的，其中 a、 b、 c 均或者是空字符串，或者是仅由字母 A 组成的字符串。
现在就请你为 PAT 写一个自动裁判程序，判定哪些字符串是可以获得“答案正确”的。

输入格式：
每个测试输入包含 1 个测试用例。第 1 行给出一个正整数 n (≤10)，是需要检测的字符串个数。接下来每个字符串占一行，字符串长度不超过 100，且不包含空格。

输出格式：
每个字符串的检测结果占一行，如果该字符串可以获得“答案正确”，则输出 YES，否则输出 NO。

参考：（C语言最详细解析）1003 我要通过！ (20 分)

#include <iostream>
#include <string.h>

using namespace std;

int main()
{
    //ios::sync_with_stdio(false);
    int n,flag1,flag2,flag3,k = 0;
    cin >> n;
    char s[101];
    for(int i = 0;i < n;i++){
        scanf("%s",s);
        int locP,locT,cntP = 0,cntT = 0,other = 0;
        int len = strlen(s);
        for(int j = 0;j < len;j++){
            //记录P位置和P的出现次数
            if(s[j] == 'P'){
                locP = j;
                cntP++;
            }
            //记录T位置和出现次数
            else if(s[j] == 'T'){
                locT = j;
                cntT++;
            }
            //出现了其他字符
            else if(s[j] != 'A'){
                other++;
            }
        }
        //判断
        //出现了其他字符||P和T分别出现了不止一次||PT之间没有A
        if(other != 0 || cntP != 1 || cntT != 1 ||locT - locP <= 1){
            cout << "NO" << endl;
            continue;
        }
        //判断A的情况
        int x = locP;//在P之前的A的个数
        int y = locT - locP - 1;//P和T之间A的个数
        int z = len - locT - 1;//在T之后的A的个数
        if(x * y == z){ //符合条件
            cout << "YES" << endl;
        }
        else   cout << "NO" << endl;

    }
    return 0;
}