“答案正确”是自动判题系统给出的最令人欢喜的回复。本题属于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
【思路】
第二个条件说明PAT左右可以有任意A但是必须是数量相同的,第三个条件说明每当中间增加一个A,那么右面增加和左面数量相同的A,结合第二个条件,进一步说明PAT右面的A的数量是左面A数量的整数倍,并且这个数就是PT之间A的个数。
【C++程序】
---------------
#include <iostream>
#include <string>
using namespace std;
int main() {
int n;
cin >> n;
while(n--) {
string str;
cin >> str;
// P之前第一个不是字符'A'的位置
string::size_type posA = str.find_first_not_of('A');
if(posA == string::npos || str[posA] != 'P') {
cout << "NO" << endl;
continue;
}
// PT之间第一个不是字符'A'的位置
string::size_type posPT = str.find_first_not_of('A', posA + 1);
if(posPT == string::npos || str[posPT] != 'T' || (posPT == posA + 1)) {
cout << "NO" << endl;
continue;
}
// T之后第一个不是字符'A'的位置
string::size_type posT = str.find_first_not_of('A', posPT + 1);
if(posT != string::npos) {
cout << "NO" << endl;
continue;
}
// T之后A的个数 = P之前A的个数 ×PT之间A的个数
if((str.length() - posPT - 1) == posA * (posPT - posA - 1))
cout << "YES" << endl;
else
cout << "NO" << endl;
}
return 0;
}