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。

输入样例：
10
PAT
PAAT
AAPATAA
AAPAATAAAA
xPATx
PT
Whatever
APAAATAA
APT
APATTAA

输出样例：
YES
YES
YES
YES
NO
NO
NO
NO
NO
NO

二、解析

这道题完完全全就是一个数学题。第一个条件好理解，字符串只能有P A T三种字符。第二个条件说xPATx是正确的，x是空串或者只含A的字符串。我么先假设PAT是正确的，那么根据第三个条件，如果aPbTc是正确的，这里a、c都是空串，b是A，那么aPbATca正确，即PAAT是正确的。我们再将PAAT看成是aPbTc，即a、c是空串，b是AA，那么根据第三个条件，PAAAT也是正确的。所以我们得出一个结论，PAAAAA…T是正确的，也就是P和T之间有好多好多个A这种是正确的。
再看，xPATx如果是APAT，它是正确的，那么aPbTc即a是A，b是A，c是A，那么aPbATca即APAATAA是正确的。再根据第三个条件推理，APAAATAAA也是正确的。
根据上面推理，我们就可以得到如下结论。设PAT前的A的数量为K1，PT中间A的数量为K2，PT后A的数量为K3，那么当K1*K2==K3，且K2不为0时，字符串是正确的，否则错误。

三、代码

#include <iostream>
#include <cstring>
#include <algorithm>

using namespace std;

bool judge(string s){
    //先判断字符串是不是只含有P A T
    for(int i = 0;i<s.size();i++)
    {
        if(s[i]!='P'&&s[i]!='A'&&s[i]!='T'){
            return false;
        }
    }

    int k1=0,flag1=0;//k1表示P前A的个数，flag1表示是否第一次碰到了P
    int k2=0,flag2=0;//k2表示P和T之间的A的个数，flag2表示是否第一次碰到了T
    int k3=0;//k3表示PAT后的A的个数
    int kp=0,kt=0; //记录P和T的个数 合法的字符串应该只有一个P和一个T
    for(int i = 0;i<s.size();i++){
    if(flag1==0&&flag2==0){ //还没碰到P和T，那么碰到的A一定是P之前的A
        if(s[i]=='A')
            k1++;
    }
    if(s[i]=='P')
    {
        kp++;
        flag1=1;
    }
    if(flag1==1&&flag2==0){//碰到过P，但是没有碰到过T，那么碰到的A一定在P和T之间
        if(s[i]=='A')
            k2++;
    }
    if(s[i]=='T')
    {
        kt++;
        flag2=1;
    }
    if(flag1==1&&flag2==1){ //P和T都碰到过，说明再碰到的A一定在PAT的后面
        if(s[i]=='A')
            k3++;
    }
    }
    if(kp!=1||kt!=1||k1+k2+k3==0)//P和T只有一个  PT不正确
        return false;
    if(k1*k2==k3&&k2!=0)//要注意k2不能等于0 PT中间至少有一个A
        return true;
    else
        return false;

}

int main()
{
    int n;
    cin>>n;
    while(n!=0){
        string s;
        cin>>s;
        if(judge(s)==true)
            cout<<"YES"<<endl;
        else if(n==1)
            cout<<"NO";
        else
            cout<<"NO"<<endl;
        n--;
    }


    return 0;
}