题目:
“答案正确”是自动判题系统给出的最令人欢喜的回复。本题属于PAT的“答案正确”大派送 —— 只要读入的字符串满足下列条件,系统就输出“答案正确”,否则输出“答案错误”。
得到“答案正确”的条件是:
1. 字符串中必须仅有P, A, T这三种字符,不可以包含其它字符;
2. 任意形如 xPATx 的字符串都可以获得“答案正确”,其中 x 或者是空字符串,或者是仅由字母 A 组成的字符串;
3. 如果 aPbTc 是正确的,那么 aPbATca 也是正确的,其中 a, b, c 均或者是空字符串,或者是仅由字母 A 组成的字符串。
输入格式: 每个测试输入包含1个测试用例。第1行给出一个自然数n (<10),是需要检测的字符串个数。接下来每个字符串占一行,字符串长度不超过100,且不包含空格。
输出格式:每个字符串的检测结果占一行,如果该字符串可以获得“答案正确”,则输出YES,否则输出NO。
输入样例:8 PAT PAAT AAPATAA AAPAATAAAA xPATx PT Whatever APAAATAA输出样例:
YES YES YES YES NO NO NO NO
题目说了很多、很长。但是仔细观察后发现,其实合法字符串的条件是:
1、只含有PAT三个字符,且P、T仅有各有一个,A可有一个或多个;
2、T后A的个数与P前A的个数之商等于PT中间A的个数。
由此可以简单写出AC代码:
#include<iostream>
using namespace std;
bool judge(const string &);
int main()
{
int n;
string str;
cin>>n;
for(;n--;)
{
cin>>str;
cout<<(judge(str)?"YES":"NO")<<endl;
}
return 0;
}
bool judge(const string &str)
{
const char* cstr = str.c_str();
int lP=-1,lT=-1;
for(unsigned int i=0;i<str.length();i++)
{
if(cstr[i]=='A')
continue;
if(cstr[i]=='P'&&lP==-1)
{
lP=i;
continue;
}
if(cstr[i]=='P'&&lP!=-1)
return false;
if(cstr[i]=='T'&&lT==-1)
{
lT = i;
continue;
}
if(cstr[i]=='T'&&lT!=-1)
return false;
return false;
}
int front = lP,mid = lT-lP-1,back = str.length()-1-lT;
if(lP==-1||lT==-1||mid==0) return false;
if(front==0&&back==0) return true;
if(back%front==0&&back/front==mid) return true;
return false;
}