PTA 乙级 1003 我要通过！ (20 分) C++

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。

输入样例：

9
PAT
PAAT
AAPATAA
AAPAATAAAA
xPATx
PT
Whatever
APAAATAA
APT

输出样例：

YES
YES
YES
YES
NO
NO
NO
NO
NO

解题思路

1. 由条件1可得，输入只能含有PTA三个字母，不能包含别的字母
2. 由条件2可得，APATA,AAPATAA…也正确，注意两边的A数量要相等

如果 aPbTc 是正确的，那么 aPbATca 也是正确的，其中 a、 b、 c 均或者是空字符串，或者是仅由字母 A 组成的字符串。

3. 条件3相对而言就麻烦了一些，需要结合条件1和条件2，以及输入输出的结果，
   • 由结果PT为NO，得出b不能为空，即是PT中必须有A。
   • 结合条件2， 令a为A，b为A，c为A，APATA为true -> APAATAA为true
   • 令a为A，b为AA，c为AA，APATA-> APAAATAAA为true，继续推导可知，当a=1时，len（c） = len（b）。
   • 令a为AA，b为A，c为AA， ->AAPAATAAAA为true，仿照上面的步骤可以推出当a=2时，len（c） = 2len（b）。
   • 继续推导可以得出 len（a）*len（b） = len（c）。
4. 由APT为NO可知，P必须在T前面。
5. 由上述条件可知，P和T的个数只能有一个。

文字游戏hahaha

---

C++代码

#include<iostream>
#include<string>
using namespace std;
int main()
{
    int n,i;
    string s;
    cin>>n;
    for(i = 0;i<n;i++)
    {
        cin>>s;
        int len = s.size();
        //T是T在字符串的位置，P同理
        //tc和pc是T和P的个数 ，Asign是标记位，标记是否出现了A
        //sign是标记是否出现其他字母
        int T=-1,P=-1,tc=0,pc=0,Asign=0,sign=1;
        for(int i = 0;i<len;i++)
        {
            if(s[i]=='T'){T=i;tc++;} 
            else if(s[i]=='P') {P=i;pc++;}
            else if(s[i]=='A') {Asign=1;}
            else sign=0;
        }    
        //如果出现其他字母，如果没有出现A，如果P在T后面，如果P和T的个数不为1，如果len（a）*len（b） != len（c），则输出no
            if(sign==0||Asign==0||P>=T-1||tc!=1||pc!=1||(T-P-1)*P!=len-T-1) {
            //判断是否为最后一个，是就不需要换行符
                if(i!=n-1)
                	cout<<"NO"<<endl;
            else
             	cout<<"NO";
            }
            else {
                if(i!=n-1)
                    cout<<"YES"<<endl;
                else
                    cout<<"YES";}   
    }
    return 0;
}