1003. 我要通过！(20)

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

题目解读：3个条件是有关的，第一个条件表示必须有且只有 P、A、T 3个字符
第二个条件表示”PAT” 左右有任意相等个数个’A’都符合条件（我一开始以为两端A的个数不同也符合条件，所以一直没A过去）
第三个条件和1、2条件有关，（2）条件下，APATA AAPATAA AAAPATAAA正确，又因为 aPbTc 是正确的，那么 aPbATca 也是正确
所以 APAATAA AAPAATAAAA …也正确，
所以 APAAATAAA AAPAAATAAAAAA … 也正确
总之就是 p_a*m_a == t_a, 就是P前A的个数乘上P、T之间A的个数等于T后A的个数

#include <iostream>
#include <vector>
#include <string>
using namespace std;

int flag(string s);

int main(){

    int n;
    cin >> n;
    getchar(); // 消除回车键
    vector<string> v(n);

    for(int i = 0; i < n; ++i)
        getline(cin, v[i]); // getline() 每次读一行

    for(int i = 0; i < n; ++i)
        if(flag(v[i]))
            cout << "YES" << endl;
        else
            cout << "NO" << endl;

    return 0;
}

int flag(string s){

    int p = s.find('P'), t = s.find('T');
    int p_a = 0, m_a = 0, t_a = 0; // 通过P、T把字符串分成3部分，这3个数字分别代表3部分'A'的个数

    if(p == string::npos && t == string::npos) // 找不到到P和T
  // size_type string::npos 表示找一个字符没找到，常和find()搭配
        return 0;
    if(p+1 == t) // P、T之间没有A，不符合条件
        return 0;
    if(p == 0 && t == s.size()-1){ // 两端没有'A'
        for(int i = 1; i < t; ++i)
            if(s[i] != 'A')
                return 0;

        return 1;
    }
// P、T两端有A
    for(int i = 0; i < p; ++i){
        if(s[i] != 'A')
            return 0;
        else
            p_a++;
    }

    for(int i = p+1; i < t; ++i){
        if(s[i] != 'A')
            return 0;
        else
            m_a++;
    }

    for(int i = t+1; i < s.size(); ++i){
        if(s[i] != 'A')
            return 0;
        else
            t_a++;
    }

    if(t_a == p_a*m_a)
        return 1;
    else
        return 0;

}