【PAT】1003 我要通过！ (20 分)

题目链接

菜鸟做法

#include <iostream>
#include <bits/stdc++.h>
using namespace std;

int main()
{
    int n;
    string a;
    cin>>n;
    //观察之后，会发现一个规律
    //前面的A的数量×中间的A的数量=最后的A的数量
    while(n--){
        cin>>a;
        //除了PAT外没有别的字母
        int x=0,y=0,z=0;
        int flag=0;//0代表前面，1代表中间，2代表后面
        int ifout=0;
        for(int i=0;i<a.length();i++){
            if(a[i]=='P'&&flag==0){
                flag=1;
                continue;
            }else if(a[i]=='T'&&flag==1){
                flag=2;
                continue;
            }
            if(a[i]!='A'){
                ifout=1;
                break;
            }else{
                if(flag==0){
                    x++;
                }else if(flag==1){
                    y++;
                }else{
                    z++;
                }
            }
        }
        //一共有四个检查点，我就漏了一个┭┮﹏┭┮，居然把第③个给忘了
        //① 只存在PAT
        //② PT之间一定有A
        //③ PAT必须都存在
        //④ 前面A的个数×中间A的个数=后面A的个数
        if(x*y==z&&y!=0&&ifout!=1&&flag==2){
            cout<<"YES"<<endl;
        }else{
            cout<<"NO"<<endl;
        }

    }
    return 0;
}

大佬做法

#include <iostream>
#include <map>
using namespace std;
int main() {
    int n, p = 0, t = 0;
    string s;
    cin >> n;
    for(int i = 0; i < n; i++) {
        cin >> s;
        map<char, int> m;
        for(int j = 0; j < s.size(); j++) {
            m[s[j]]++;
            if (s[j] == 'P') p = j;
            if (s[j] == 'T') t = j;
        }
        if (m['P'] == 1 && m['A'] != 0 && m['T'] == 1 && m.size() == 3 && t-p != 1 && p * (t-p-1) == s.length()-t-1)
            printf("YES\n");
        else
            printf("NO\n");
    }
    return 0;
}