1003 我要通过！(C++版)

题目如下：

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

int  juage(string str)
{
	int n_p = 0, n_t = 0,n_a=0;
	for (int j = 0; j < str.length(); j++)
	{
		//如果某个字符不是P A T三个字符中的一个，返回0
		if (str[j] != 'P' && str[j] != 'T' && str[j] != 'A')
			return 0;
		//要求必须含有PAT，且计算P&T个数，要求P&T只出现一次，个数为1，否则返回0
		if (str[j] == 'P')
			n_p++;
		if (str[j] == 'T')
			n_t++;
		if (str[j] == 'A')
			n_a++;
	}
	if (n_p != 1 || n_t != 1||n_a<1)
		return 0;

	//P必须在T前面，否则返回0
	int Pp, Pt;
	Pp = str.find('P');
	Pt = str.find('T');
	if (Pp > Pt)
		return 0;

	//A需满足两个条件：1）位置：x1 P x2 T x3; 2）x1*x2=x3
	int x1 = 0, x2 = 0, x3 = 0;
	for (int j = 0; j < str.length(); j++)
	{
		if (j < Pp)  
			x1++;
		if (j > Pp && j < Pt)
			x2++;
		if (j > Pt)
			x3++;
	}
	if (x1 * x2 != x3)
		return 0;
	else
		return 1;
}
int main()
{
	int n=0;
	cin >> n;
	string str;
	for (int i = 0; i < n; i++)
	{
		cin >> str;
		if (juage(str))
			cout << "YES" << endl;
		else
			cout << "NO" << endl;
	}
	return 0;
}