PAT乙级——1003 我要通过！ (20分)（C++）

悠悠 2023-07-05 10:03 1阅读 0赞

题目链接：[https://pintia.cn/problem-sets/994805260223102976/problems/994805323154440192][https_pintia.cn_problem-sets_994805260223102976_problems_994805323154440192]

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

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

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

### 思路： ###

之前看半天没明白这一题到底是什么意思.......看了各个大佬的blog才明白，给出的字符串首先必须要有PAT，然后给出来的3那一长串的意思是，P之前的A的个数为n1，PT之间的A的个数为n2，T之后的A的个数为n3，要满足n3=n2\*n1就可以了。于是写一个评判标准函数，字符串为传入参数，如果遍历字符串既没有P，也没有A，也没有T，那么直接不考虑。然后开始记录P，A，T三个字母的个数，如果P不止一个，或者T不止一个，或者A没有，都算错（A可以有多个）。最后一种情况是通过调用find()函数确定P和A和T的位置，首先P一定要在T之前，其次统计n1,n2,n3，最后看是否满足n3=n1\*n2即可。

### 代码： ###

#include<iostream>
    #include<string>
    #include<algorithm>
    using namespace std;
    bool judge(string s)///评判函数
    {
        int count_p=0,count_a=0,count_t=0;///PAT三种字母的个数
        int pos_p=0,pos_a=0,pos_t=0;///PAT各数字的位置
        int num1=0,num2=0,num3=0;///**1**P**2**T**3**,分别表示位置123的A的个数，满足n3=n1*n2就满足条件
        for(int i=0;i<s.length();i++)
        {
            if(s[i]!='P'&&s[i]!='A'&&s[i]!='T')///首先要是每个字母每一个是PAT，直接pass
                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_a<1||count_t!=1)
            return false;///没有A，直接pass
        pos_p=s.find('P');///用到函数找到相应字母的位置
        pos_a=s.find('A');
        pos_t=s.find('T');
        if(pos_p>=pos_t)///P必须在T的前面，保证顺序
            return false;
        for(int j=0;j<s.length();j++)
        {
            if(j<pos_p&&s[j]=='A')
                num1++;
            else if(j>pos_p&&j<pos_t&&s[j]=='A')
                num2++;
            else if(j>pos_t&&s[j]=='A')
                num3++;
        }
        if(num3==num1*num2)
            return true;
        else
            return false;
    }
    int main()
    {
        int n;
        string s;
        cin>>n;
        for(int i=0;i<n;i++)
        {
            cin>>s;
           if(judge(s))
            cout<<"YES"<<endl;
           else
            cout<<"NO"<<endl;
        }
        return 0;
    }

[https_pintia.cn_problem-sets_994805260223102976_problems_994805323154440192]: https://pintia.cn/problem-sets/994805260223102976/problems/994805323154440192