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。

输入样例：

8
PAT
PAAT
AAPATAA
AAPAATAAAA
xPATx
PT
Whatever
APAAATAA

输出样例：

YES
YES
YES
YES
NO
NO
NO
NO

---

分析：

  看到题目第一遍，感觉特别简单，不就是判断字符串嘛。然而看到输入样例和输出样例，心中一万匹***跑过，好像不对。思考良久没有什么思路，于是打开柳神博客～
  一句话概括：

能有一个P一个T，中间末尾和开头可以随便插入A。
但是必须满足开头的A的个数 * 中间的A的个数 = 结尾的A的个数，而且P和T中间不能没有A。

代码：

#include <iostream>
#include <map>
#include <string>
using namespace std;
int main()
{
    // 记录p和t的位置
    int p = 0, t = 0;
    string str;
    int num;
    while(cin>>num)
    {
        for(int i = 0 ; i < num ; i ++)
        {
            cin>>str;
            map<char, int> content ;
            for(int j = 0 ; j < str.length() ; j ++)
            {
                if(str[j] == 'P') p = j;
                if(str[j] == 'T') t = j;
                // 记录字符串中其他字符个数
                content[str[j]]++;
            }
            // 约束p的个数为1，t的个数为1，a的个数任意，只包含三个字母，p和t之间至少有一个字母（t-p != 1）
            // a - b - 1 : a和b之间有多少个位置，多 - 1 是代表把a自己减掉
            if(content['P'] == 1
            && content['T'] == 1
            && content['A'] != 0
            && content.size() == 3
            && t-p != 1
            && p * (t - p - 1) == str.length() - t - 1)
            {
                cout<<"YES"<<endl;
            } else {
                cout<<"NO"<<endl;
            }
        }
    }
}