PAT basic 1003

#include<iostream>
#include<string>
using namespace std;
int main()
{
    int n;
    cin >> n;
    while(n--)
    {
        string s;
        cin >> s;
        int a1 = 0,a2 = 0,a3 = 0;
        int i = 0;
        int flag = 0;
        for(int j = 0;j < s.length();j++)
        {
            if((s[j] != 'P')&&(s[j] != 'A')&&(s[j] != 'T'))
            flag = 1;
        }
        while(s[i] == 'A')
        {
            i++;
            a1++;
        }
        if(s[i] == 'P')
        {
            if(s[i+1] == 'P')
            flag = 1;
            i++;
        }
        while(s[i] == 'A')
        {
            i++;
            a2++;
        }
        if(s[i] == 'T')
        {
            if(s[i+1] == 'T')
            flag = 1;
            i++;
        }
        while(s[i] == 'A')
        {
            i++;
            a3++;
        }
        if(i != s.length())
        flag = 1;
        cout << a1 << " " << a2 << " " << a3 <<" "<< flag <<endl;
        if((a3 ==  (a1 * a2 ))&& (a2 != 0)&&(!flag))    
        cout << "YES" <<endl;
        else
        cout << "NO" <<endl;
    }
    return 0;
}
/* 妈的智障 测试点二总是过不去
    不知道是哪里有错误 
    不知道是哪一个点没有考虑到
    其实这样的解法也是非常无聊的啊啊、 
 */