1003

1003 我要通过！(20)（20 分）

作者: CHEN, Yue
单位: PAT联盟
时间限制: 400ms
内存限制: 64MB
代码长度限制: 16KB

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

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

1. 字符串中必须仅有P, A, T这三种字符，不可以包含其它字符；\

   任意形如 xPATx 的字符串都可以获得“答案正确”，其中 x 或者是空字符串，或者是仅由字母 A 组成的字符串；\
   如果 aPbTc 是正确的，那么 aPbATca 也是正确的，其中 a, b, c 均或者是空字符串，或者是仅由字母 A 组成的字符串。

现在就请你为PAT写一个自动裁判程序，判定哪些字符串是可以获得“答案正确”的。

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

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

输入样例：

8
PAT
PAAT
AAPATAA
AAPAATAAAA
xPATx
PT
Whatever
APAAATAA

输出样例：

YES
YES
YES
YES
NO
NO
NO
NO

---

分析：（1.）任意xPATx（x为题意所述字符串）输出YES；（2.）对可输出YES字符串 aPbTc ，aPbATca也输出YES。
需要理解的点是（2.）是具有自包含关系的，也就是说，可以将任意符合（1.）的字符串进行任意次（2.）操作，产生的字符串也输出YES。

设x中有n1个A，条件（1.）中字符串符合

n1*A+P+1*A+T+n1*A                                                          （1.1）

条件（2.）中的映射关系是P和T之间每增加1个A，T后面要增加n1个A；设P和T之间有n2个A（n2>0）。字符串符合

n1*A+P+A+(n2-1)*A+T+n1*A+(n2-1)*n1*A                                       （1.2）

化简后可得

n1*A+P+n2*A+T+n2*n1*A                                                      （1.3）

当n2=1时，（1.3）与（1.1）相等。

因此，本题转化为求P前，P与T之间，T后的A的数量关系。

---

#include<iostream>
using namespace std;

bool check(string str)
{
    int q=str.length()-1,p=0;
    while(str[p]!='P'&& p<q)
    {
        if(str[p]!='A')
        {
            return false;
        }
        p++;
    }
    while(str[q]!='T' && p<q)
    {
        if(str[q]!='A')
        {
            return false;
        }
        q--;
    }
    for(int i=p+1;i<q;i++)
    {
        if(str[i]!='A')
            return false;
    }
    if(p*(q-p-1)!=str.length()-q-1 || p+1==q)
    {
        return false;
    }
    else
    {
        return true;
    } 
}


int main()
{
    int n;
    string st;
    cin>>n;
    for(int i=0;i<n;i++)
    {
        cin>>st;
        if(check(st))
        {
            cout<<"YES"<<endl;
        }
        else
        {
            cout<<"NO"<<endl;
        };
    }   
    return 0;
}