一、题目描述
答案正确”是自动判题系统给出的最令人欢喜的回复。本题属于 PAT 的“答案正确”大派送 —— 只要读入的字符串满足下列条件,系统就输出“答案正确”,否则输出“答案错误”。
得到“答案正确”的条件是:
1.字符串中必须仅有 P、 A、 T这三种字符,不可以包含其它字符;
2.任意形如 xPATx 的字符串都可以获得“答案正确”,其中 x 或者是空字符串,或者是仅由字母 A 组成的字符串;
3.如果 aPbTc 是正确的,那么 aPbATca 也是正确的,其中 a、 b、 c 均或者是空字符串,或者是仅由字母 A 组成的字符串。
现在就请你为 PAT 写一个自动裁判程序,判定哪些字符串是可以获得“答案正确”的。
二、方法简介
我们可以仔细分析一下得到正确答案的三个条件,由1和2我们可以得到,要使输入的字符串是正确答案,这个字符串必须只能含有一个P和一个T,并且A的数量不能少于1
那么难点就在第三个条件的分析,第三个条件到底是什么意思呢,我们看一下第三个条件给的例子
如果aPbTc是正确的,那么aPbATca也是正确的
我们这里给出两个具体例子来帮助大家分析这个条件的意思(假设a、b、c都是A)
1.APATA---APAATAA
a=A, b=A, c=AA
2.AAPATAA---AAPAATAAAA
a=AA, b=A, c=AA
观察上面两个例子我们发现,我们把T之后A的数量记为n3,P和T之间的A的数量记为n2,P之前的A的数量记为n1,则有n3=n1*n2,这个就是通过条件3得到的结论,大家也可以之间去尝试多试几个,去理解这个等式是怎么来的
三、代码
#include<iostream>
#include<string>
using namespace std;
bool judge(string s)
{
int count_p = 0,count_a = 0,count_t = 0;
int n1 = 0,n2 = 0,n3 = 0;
for(int i=0;i<s.length();i++)
{
if(s[i]!='P' && s[i]!='A' && s[i]!='T') return false;
if(s[i]=='P') count_p++;
else if(s[i]=='A') count_a++;
else if(s[i]=='T') count_t++;
}
if(count_p!=1 || count_t!=1 || count_a<1) return false;
int position_p = s.find('P');//取P的位置
int position_t = s.find('T');//取T的位置
if(position_p >= position_t) return false;
for(int i=0;i<s.length();i++)
{
if(i<position_p && s[i]=='A') n1++;
else if(i>position_p && i<position_t && s[i]=='A') n2++;
else if(i>position_t && s[i]=='A') n3++;
}
if(n3==n1*n2) return true;
else return false;
}
int main()
{
int n;
cin>>n;
string s;
for(int i=0;i<n;i++)
{
cin>>s;
if(judge(s)) cout<<"YES"<<endl;
else cout<<"NO"<<endl;
}
return 0;
}