pat 乙 1003. 我要通过！

关键理解前面乘以中间的个数等于后面的个数

#include<iostream>
#include<string>
using namespace std;


int main()
{
	int n;
	cin>>n;
	
	string tmp;
	
	while(n--)
	{
		cin>>tmp;
		
		int index_p,index_t;
		int count_p,count_t,count_a;
		count_p=count_t=count_a=0;
		
		
		for(int i=0;i<tmp.size();i++)
		{
			if(tmp.at(i)=='P')
			{
				index_p=i;
				count_p++;
			}
			else if(tmp[i]=='A')
			{
				count_a++;
			}
			else if(tmp[i]=='T')
			{
				count_t++;
				index_t=i;
			}
		}
		
		if(count_p+count_a+count_t!=tmp.size() || count_p!=1 || count_t!=1 || count_a==0||
		index_p*(index_t-index_p-1)!=tmp.size()-index_t-1
		)
		
		cout<<"NO"<<endl;
		
		else cout<<"YES"<<endl;
	}
	
	
	return 0;
 }