首先分析三个条件:
- 必须仅包含P A T三个字符,说明一旦出现其他字符便会被判定为错误。
- 形如xPATx,x有两种选择,一是空字符,对应的是输入样例的PAT,说明正确,二是x为A,这时候可以表示为APATA,同样x还可以是AA,表示为AAPATAA,以此类推,可以找出规律,x可以为任意个A,也就是中间有一个A,左右两边A的个数相同则会被判定为正确。
- 如果aPbTc是正确的,若a,c为空,b为A,对应的是PAT,此时成立;若a,c为空,b为AA,对应的是PAAT,此时成立,依次类推,PAAAT,PAAAAT也是正确的,若a,c为2个a,b为1个a时,对应输入样例的AAPATAA,这个时候也是正确的,若a,c为2个a,b也为2个a时,对应的时AAPAATAAAA,这个时候可以总结规律为:a,c一致,b为任意个A都是会被判定为正确。
第3个条件中提到,若aPbTc是正确的,那么aPbATca也是正确的,也就是假设a,c为一致,b为任意A时成立,举个例子,a=AA,c=AA,b=A时,对应AAPAATAAAA,这时候是成立的,由此可见同样满足上述的规律。
**总结:**找到上述中符合条件的字符串有:
PAT
PAAT
PAAAT
APATA
AAPATAA
AAPAATAAAA
…
可以看到他们统一满足于一个规律:P和T都只有一个,且P在T的前面,P和T中间不能为空,并且P前面A的个数*中间A的个数=T后面A的个数,另外就是不可以出现其他字符。
找到规律之后,就可以编写代码了,我这里利用到的队列,因为队列有一个先进先出的规则,那么就是可以按顺序判断队列中的每个字符串是否满足条件,用的方法稍微暴力了一点,不过比较好理解。代码如下:
#include<iostream>
#include<string>
#include<queue>
using namespace std;
int main(){
int n;
string s;
cin >> n;
queue<string> S;
for(int i=0; i<n; i++){
cin >> s;
S.push(s); //将输入的每个字符输入进队列中,方便按照顺序进行访问并判断
}
int p=0, t=0, a=0, b=0, c=0, d=0;
int p1=0, t1=0, a1=0, b1=0, c1=0, d1=0; //这里是为了统计P A T满足条件出现的次数,为了方便判断是否符合:P前面A的个数*中间A的个数=T后面A的个数,每个字符的含义都在下面的判断条件中给出。
for(int i=0; i<S.size();i++){
for(int j=0; j<S.front().length();j++){
if(S.front()[j]=='P'||S.front()[j]=='T'||S.front()[j]=='A'){
if(S.front()[j]=='P'){ //统计P的次数
++p;
}
if(S.front()[j]=='T'&& (p!=0)){ //统计P之后T的个数
++t;
}
if(S.front()[j]=='A'&& p==0){ //统计A在P前面的A的个数
++a;
}
if(S.front()[j]=='A'&& p==1 && t==0){ //统计A在P和T中间的A的个数
++b;
}
if(S.front()[j]=='A'&& p==1 && t==1){ //统计A在P和T后面的A的个数
++c;
}
}
else{
++d; //这里是统计出现其他字符的情况,如果有其他字符,则+1
}
}
p1 = p;
p = 0;
t1 = t;
t = 0;
a1 = a;
a = 0;
b1 = b;
b = 0;
c1 = c;
c = 0;
d1 = d;
d=0; //这里每次统计完之后都要记得清0,防止访问下一个字符串的时候把上一个字符串的信息累加进去
if(p1==1 && t1==1 && a1*b1==c1 && b1!=0 && d1==0) //这里的d1是表示其他字符,只有其他字符出现次数为0时才正确
{
cout << "YES" <<endl;
}
else{
cout << "NO" << endl;
}
S.push(S.front());
S.pop();
}
return 0;
}