PAT乙级 1003 我要通过！ (数学/字符串/映射)

前面A的个数乘以中间A的个数等于后面A的个数
参考 柳婼姐姐的博客

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