PTA刷题之路---乙类1003 我要通过！

一、题目描述

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

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

1.字符串中必须仅有 P、 A、 T这三种字符，不可以包含其它字符；
2.任意形如 xPATx 的字符串都可以获得“答案正确”，其中 x 或者是空字符串，或者是仅由字母 A 组成的字符串；
3.如果 aPbTc 是正确的，那么 aPbATca 也是正确的，其中 a、 b、 c 均或者是空字符串，或者是仅由字母 A 组成的字符串。

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

二、方法简介

我们可以仔细分析一下得到正确答案的三个条件，由1和2我们可以得到，要使输入的字符串是正确答案，这个字符串必须只能含有一个P和一个T，并且A的数量不能少于1

那么难点就在第三个条件的分析，第三个条件到底是什么意思呢，我们看一下第三个条件给的例子

如果aPbTc是正确的，那么aPbATca也是正确的

我们这里给出两个具体例子来帮助大家分析这个条件的意思（假设a、b、c都是A）

1.APATA---APAATAA
  a=A, b=A, c=AA
2.AAPATAA---AAPAATAAAA
  a=AA, b=A, c=AA

观察上面两个例子我们发现，我们把T之后A的数量记为n3，P和T之间的A的数量记为n2，P之前的A的数量记为n1，则有n3=n1*n2，这个就是通过条件3得到的结论，大家也可以之间去尝试多试几个，去理解这个等式是怎么来的

三、代码

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

bool judge(string s)
{
    int count_p = 0,count_a = 0,count_t = 0;
    int n1 = 0,n2 = 0,n3 = 0;
    for(int i=0;i<s.length();i++)
    {
        if(s[i]!='P' && s[i]!='A' && s[i]!='T') return false;
        if(s[i]=='P') count_p++;
        else if(s[i]=='A') count_a++;
        else if(s[i]=='T') count_t++;
    }

    if(count_p!=1 || count_t!=1 || count_a<1) return false;

    int position_p = s.find('P');//取P的位置
    int position_t = s.find('T');//取T的位置

    if(position_p >= position_t) return false;

    for(int i=0;i<s.length();i++)
    {
        if(i<position_p && s[i]=='A') n1++;
        else if(i>position_p && i<position_t && s[i]=='A') n2++;
        else if(i>position_t && s[i]=='A') n3++;
    }

    if(n3==n1*n2) return true;
    else return false;
}

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