PAT--1003 我要通过！

“答案正确”是自动判题系统给出的最令人欢喜的回复。本题属于 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

分析
这题目第一眼看上去挺难的，其实最主要还是因为第一遍看很难看懂，第二遍仔细分析一下每一个条件：

1.字符串中必须仅有 P、 A、 T这三种字符，不可以包含其它字符；

这个条件很简明，就是如果这个字符串包含有P、A、T以外的字符，就是非法的

2.任意形如 xPATx 的字符串都可以获得“答案正确”，其中 x 或者是空字符串，或者是仅由字母 A 组成的字符串；

首先看一下这个字符串：xPATx，意思也比较清晰，就是你中间只要是PAT，然后两边是一个由N个A组成的对称结构（N可以为0），就是合法的

这时候其实就可以看出一些规律了，不管P和T，整体的A分布是按照 N–1--N 的规律分布的（中间两杠是P和T），再来看第三个条件

3.如果 aPbTc 是正确的，那么 aPbATca 也是正确的，其中 a、 b、 c 均或者是空字符串，或者是仅由字母 A 组成的字符串。

题目上说的很明显， a、 b、 c 均或者是空字符串，或者是仅由字母 A 组成的字符串。
那我们就把a\b\c看作是由a个A、b个A、c个A组成的字符串（a、b、c可以为0）
则这时候A个数的分布是按照 a–b--c 分布的
再看字符串 aPbATca ，它的分布是 a–b+1–c+a
如果在第二部没有看出规律，到这一步仔细思考一下就能看出：b加了一，c就加了一个a；这不就是乘法吗
在第二步中有：N * 1 = N
而在第三步中：如果 aPbTc （a * b = c）是正确的，那么aPbATca 也是正确的（a * b+1 = c+a）

总结：最后就是P左边A的个数 * P和T中间A的个数 == T右边A的个数

#include<iostream> 
#include<cstring>
using namespace std;

int main(){
	int N;	cin>>N;
	char test[10][101];
	for(int i=0;i<N;i++){
		cin>>test[i];
		int size=strlen(test[i]);
		//cout<<size;
		int FR=0,MI=0,BK=0;		//记录P前，P,T中，T后 A的个数 
		int p=0,a=0,t=0;		//记录p,a,t的个数(题目要求p,t只能是1个,a不能为0) 
		int now=1;				//当前位置：1前  2中  3后 
		for(int j=0;j<size;j++){
			if(test[i][j]=='A'){
				a++;
			}
			else if(test[i][j]=='P'){
				p++;
				now=2;
			}
			else if(test[i][j]=='T'){
				t++;
				now=3;
			}
			else{		//出现了除p,a,t之外的字符，把BK置0 
				BK=0;
				break;
			}
			switch(now){
				case 1:
					FR++;
					break;
				case 2:
					MI++;
					break;
				case 3:
					BK++;
					break;
			}
		}
		if(a && p==1 && t==1 && FR*(MI-1)==(BK-1)){
			cout<<"YES"<<endl;
		}
		else{
			cout<<"NO"<<endl;
		}
	}
	return 0;
}