PAT.B1003 我要通过！

返回目录

样例（可复制）

8
PAT
PAAT
AAPATAA
AAPAATAAAA
xPATx
PT
Whatever
APAAATAA

样例输出

YES
YES
YES
YES
NO
NO
NO
NO

注意点

1. 若使用scanf()或cin>>输入后需要用getchar()吸收掉换行符‘\n’,还可以用本题"%d%*c"的方式吸收掉
2. 本题是个数学问题，思路是：先找到P和T字符的位置，得到P之前的，P和T之间的，T之后的A字符的个数a1,a2,a3,当a1==a3-a1*(a2-1)&&a2>=1时得到YES，否则若出现非PAT字符或PT字符位置不对等情况则输出NO

#include<bits/stdc++.h>
using namespace std;
int main(){
    int n;
    scanf("%d%*c",&n);
    string a;
    while(n--){
        getline(cin,a);
        int pp=-1,tt=-1;
        for(int i=0;i<a.size();i++)
            if(a[i]=='P'&&pp==-1)
                pp=i;
            else if(a[i]=='T'&&tt==-1)
                tt=i;
            else if(a[i]!='A'){
                pp=-2;
                break;
            }
        if(pp>-1&&tt>-1&&pp<tt-1&&pp*(tt-pp-1)==a.size()-1-tt)
            printf("YES\n");
        else
            printf("NO\n");
    }
    return 0;
}