PAT-B1003. 我要通过！(20)（数学）

#include <iostream>
#include <cstring>
#include <cstdio>
#include <cstdlib>
using namespace std;

#define MAX 100+10
char str[MAX];

int main()
{
//freopen("in.txt", "r", stdin);
//freopen("out.txt", "w", stdout);
int N;
scanf("%d", &N);
getchar();
while (N--)
{
gets(str);
int num_p = 0, num_a = 0, num_t = 0;
int loc_p, loc_t, len = strlen(str), f = 0;
for (int i = 0; i < len; i++)
{
if (str[i] == 'P')
{
num_p++;
loc_p = i;
}
else if (str[i] == 'A')
num_a++;
else if (str[i] == 'T')
{
num_t++;
loc_t = i;
}
else
f = 1;
}

if (1 == f || num_p != 1 || !num_a || num_t != 1)
printf("NO\n");
else if (loc_p >= loc_t)
printf("NO\n");
else
{
int x = loc_p, y = loc_t-loc_p-1, z = len-1-loc_t;
if (x == z-x*(y-1))
printf("YES\n");
else
printf("NO\n");
}
}
return 0;
}