1003. 我要通过！(20)--PAT乙级

1003. 我要通过！(20)

时间限制

400 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

CHEN, Yue

“答案正确”是自动判题系统给出的最令人欢喜的回复。本题属于PAT的“答案正确”大派送 —— 只要读入的字符串满足下列条件，系统就输出“答案正确”，否则输出“答案错误”。

得到“答案正确”的条件是：

1. 字符串中必须仅有P, A, T这三种字符，不可以包含其它字符；

2. 任意形如 xPATx 的字符串都可以获得“答案正确”，其中 x 或者是空字符串，或者是仅由字母 A 组成的字符串；

3. 如果 aPbTc 是正确的，那么 aPbATca 也是正确的，其中 a, b, c 均或者是空字符串，或者是仅由字母 A 组成的字符串。
现在就请你为PAT写一个自动裁判程序，判定哪些字符串是可以获得“答案正确”的。

输入格式： 每个测试输入包含1个测试用例。第1行给出一个自然数n (<10)，是需要检测的字符串个数。接下来每个字符串占一行，字符串长度不超过100，且不包含空格。

输出格式：每个字符串的检测结果占一行，如果该字符串可以获得“答案正确”，则输出YES，否则输出NO。
输入样例：

8
PAT
PAAT
AAPATAA
AAPAATAAAA
xPATx
PT
Whatever
APAAATAA

输出样例：

YES
YES
YES
YES
NO
NO
NO
NO

分析：“答案正确”的字符串只含'P'，'A'，'T'且'P'前的字符'A'的个数num1，'P'与'T'间的字符'A'的个数num2

'T'后字符'A'的个数num3间存在一定的关系即num1 * num2 = num3。这个关系可以从第三个条件推出。

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

string s;//字符串s
int N;//N个测试用例
int counter;//统计A的数目
int main()
{
int num1,num2,num3;
scanf("%d", &N);
while(N--)
{
cin >> s;
num1 = num2 = num3 = 0;
int m = s.find('P');
int n = s.find('T');

if(m == -1||n == -1)//不含'P'和'T',输出NO
{
printf("NO\n");
continue;
}
for(int i = 0; i < m; i++)//统计'P'前'A'的个数num1
{
if(s[i] == 'A')
num1++;
}
for(int i = m + 1; i < n; i++)//统计'P'至'T'间'A'的个数num2
{
if(s[i] == 'A')
num2++;
}
for(int i = n + 1; i < s.size(); i++)//统计'A'后'A'的个数num3
{
if(s[i] == 'A')
num3++;
}
if(num2 == 0)//'P'与'T'间不含'A',输出'NO'
{
printf("NO\n");
continue;
}
else if(m == 0&&(n == s.size()-1)&&(num2 == s.size()-2-m))//形如'PxT'形式,输出YES
{
printf("YES\n");
continue;
}
else if((num1 == m)&&(num2 == n-m-1) && (num3 == s.size()-1-n)
&& (num1 * num2 == num3))//字符串只含'P' 'A' 'T'且满足一定的数量关系
{
printf("YES\n");
continue;
}
else//其它情况输出NO
{
printf("NO\n");
continue;
}

}
}