PAT (Basic Level) Practice——1003 我要通过！

PAT (Basic Level) Practice——1003 我要通过！

题目链接：https://pintia.cn/problem-sets/994805260223102976/problems/994805323154440192

题目描述

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

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

字符串中必须仅有 P、 A、 T这三种字符，不可以包含其它字符；
任意形如 xPATx 的字符串都可以获得“答案正确”，其中 x 或者是空字符串，或者是仅由字母 A 组成的字符串；
如果 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

这道题目的难点在于对于第3个条件的判断：当aPbTc正确时，aPbATca也是正确的。
怎么理解呢，我们把所有的输入分为aPbTc的格式，一起一点点来推导：

1. PAT是正确的：a=0，b=1，c=0
2. APATA是正确的：a=1，b=1，c=1
3. APAATAA是正确的：a=1，b=2，c=2
4. APAAATAAA是正确的：a=1，b=3，c=3
5. AAPATAA是正确的：a=2，b=1，c=2
6. AAPAATAAAA是正确的：a=2，b=2，c=4
7. AAPAAATAAAAAA是正确的：a=2，b=3，c=6
8. ……
   可以看出，c=a*b，因此，第三点说的就是这个意思，明白了这一点程序就好写了。

#include<iostream>
#include<string>
using namespace std;
int main(){
	int k;
	cin>>k;
	while(k>0){
		string pat;
		int firstP, nextT=-1;
		bool flag=true;
		cin>>pat;
		for(int i=0;i<pat.length();i++){
			char s = pat[i];
			if(s=='P') firstP=i;//找出第一个P的位置 
			if(s=='T'&&nextT==-1) nextT=i;//找出第一个T的位置 
			if(s!='P'&&s!='A'&&s!='T'&&s!=' '){//如果有不合法输入直接结束 
				flag=false;
			}
		}
		if(flag){
			string s[3];//字符串分割 
			s[0]=pat.substr(0,firstP);//a
			s[1]=pat.substr(firstP+1,nextT-firstP-1);//b
			s[2]=pat.substr(nextT+1);//c
			for(int i=0;i<3;i++){//a b c 必须为空串或由A组成的串 
				for(int j=0;j<s[i].length();j++){
					if(s[i][j]!='A'){
						flag=false;
					}
				}
			}
			int n1 = s[0].length();
			int n2 = s[1].length();
			int n3 = s[2].length();
			if(n1*n2!=n3) flag=false;//这里就是第三点条件 
		}
		if(pat=="PT") flag=false;//PT是比较特殊的，PT是不正确的，单独成为一种情况 
		if(flag) cout<<"YES"<<endl;
		else cout<<"NO"<<endl;
		k--;
	}
}