PAT 1003 我要通过

这个题看似有很多条件需要考虑，第一次看到这个题的时候有点不知所措。

其实总结起来，只有两种情况能满足题目的条件，打印YES，其他的都是

打印NO。一是只有PAT，当然是可以通过的，还有一种情况就是P前面的A

乘P和T之间的A等于T后面的A。

#include <iostream>
#include <stdio.h>
#include <string.h>
#include <string>
using namespace std;

int main()
{
    int n;
    scanf("%d",&n);
    while(n--)
    {
        char ch[200];
    int idxP,idxT,len;
    scanf("%s",ch);
    string s = string(ch);
    idxP=s.find('P',0);
    idxT=s.find('T',0);
    len=s.length();
    if (idxP==string::npos||idxT==string::npos||idxP>=idxT-1)
    {
        printf("NO\n");
    }
    else
    {
        string a(s,0,idxP);
        string b(s,idxP+1,idxT-idxP-1);
        string c(s,idxT+1,len-idxT-1);
        string ans=a+b+c;
        if (ans.find_first_not_of('A')!=string::npos)
        {
            printf("NO\n");
        }
        else
        {
            if (idxP==0||len+-idxT-1==0)
            {
                printf("YES\n");
            }
            else if (idxP*(idxT-idxP-1)==len-idxT-1)
            {
                printf("YES\n");
            }
            else printf("NO\n");
        }
    }
    }

    return 0;
}