【PAT乙级】1003我要通过! (20) [数学题]

题目：

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

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

字符串中必须仅有 P、 A、 T 这三种字符，不可以包含其它字符；
任意形如 xPATx 的字符串都可以获得“答案正确”，其中 x 或者是空字符串，或者是仅由字母 A 组成的字符串；
如果 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

分析：

这道题是一道模拟类题目，考察的数学知识多一点。根据题目中的叙述，xPATx是正确的，前后字符串对称，数目相等。那么PAT正确，即x为空；如果 aPbATc正确，那么aPbAATca也正确，如此，PAAT正确，PAAAT正确，PAAA…T也正确。
aPbATa -> aPbAATaa -> aPbAAATaaa正确；（b为空）
aaPbATaa -> aaPbAATaaaa-> aaPbATaaaaaa正确；（b为空）
aaaPbATaaa -> aaaPbAATaaaaaa…（b为空）
这是由题目中一定正确的xPATx可以推出的，b如果不是空的话，那么就一定是中间结果。根据数学归纳法，前面A的个数*中间A的个数=后面A的个数。

代码：

#include <iostream>
#include <string>
using namespace std;
string pat;
int count1, count2, count3;
int num1, num2, flag = 0, number;
int main()
{
    cin>>number;
    for(int j = 0; j < number; j++){
        cin>>pat;
        flag = 0;
        count1 = 0;
        count2 = 0;
        count3 = 0;
        for(int i = 0; i < pat.length(); i++){
            if(pat[i] == 'P'){
                num1 = i;
                break;
            }
        }
        for(int i = num1 + 1; i < pat.length(); i++){
            if(pat[i] == 'T'){
                num2 = i;
            }
        }
        for(int i = 0; i < num1; i++){
            if(pat[i] != 'A'){
                flag = 1;
                break;
            }else{
                count1++;
            }
        }
        for(int i = num1 + 1; i < num2; i++){
            if(pat[i] != 'A'){
                flag = 1;
                break;
            }else{
                count2++;
            }
        }
        for(int i = num2 + 1; i < pat.length(); i++){
            if(pat[i] != 'A'){
                flag = 1;
                break;
            }else{
                count3++;
            }
        }
        if(flag == 0){
            if(count1 * count2 == count3 && count2 != 0){
                cout<<"YES"<<endl;
            }else{
                cout<<"NO"<<endl;
            }
        }else{
            cout<<"NO"<<endl;
        }
    }

    return 0;
}

优化

我的思路写出的代码用了67行才AC这道题目，但是根据某位大神的代码，可以只用22行…

代码：

找不同：

小姐姐使用了map来统计每个字母出现的个数，以出现P和T为分界点，统计了不同区域的A的数量。根据map的size可以判断是不是只有P、A、T这三个字母出现，并且要符合数量关系，以及P和T之间要有A才可以。