1003 我要通过！ (20point(s))

1003 我要通过！ (20point(s))

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

 

 思路：

条件 2:

1⃣️ 如果‘PAT’左边‘A’的个数与右边的‘A’的个数相等，那么结果正确 （左右‘A’的个数为0时，就是空串）

2⃣️ PAT左边与右边要么是空，要么是‘A’串（暗含条件就是字符串必须出现1次‘P’与1次‘T’，不能多不能少）

条件3:

0⃣️ a、b、c 均可以看作是由0～n个‘A’组成的串，例如 ：当a由0个‘A’组成时，a就是空串

1⃣️ 由 “如果 aPbTc 是正确的，那么 aPbATca 也是正确” 可知，aPbATca是否正确，需要看 aPbTc 是否正确，而aPbTc是否正确，需要判断是否符合条件1与条件2

2⃣️ 在使用条件3的时候，在原始字符串的中间增加1个‘A’，在后面增加|a|个‘A’

也就是说，如果原始字符串正确，那么中间增加1个‘A’，后面增加|a|个‘A’，结果也是正确的

3⃣️ 当我们判断是否成立的时候，可以逆向思考：

如果给你一个形如aPbATca字符串，如果消去后面|a|个‘A’，和减少中间1个‘A’后，符合条件2，那么aPbATca就是正确的

记左、中、右 ‘A’ 的个数分别为x、y、z（空字符串可以看作‘A’的个数为0）

 

逆向循环执行条件3，相当于

z=z-x（消去后面|a|个‘A’）,

y=y-1（减少中间1个‘A’）,

x=x（执行条件3，左边‘A’的个数不变）;

 

我们逆向循环执行条件3，

如果(y==1，x!=z ) 那么不符合条件2(左右‘A’个数相等)，错误

如果(x==z && y==1) 那么符合条件2

否则 继续逆向执行条件3

 

由于当y的值变成1时，执行条件3 一共 y-1次，

所以 z=(y-1) * (z-x) 

也就是说，如果 x == z - (y-1) * x ，那么结果正确，否则错误

#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include<algorithm>
#include<vector>
#include<cstring>
#include<string>
using namespace std;


int main() {
	int n;
	scanf("%d",&n);
	char str[108];
	int cntp, pPos, cntt, tPos, other; // cnt用于计算个数，Pos用于记录下标
	int A[108];//保存区间[0~i]中‘A’的个数

	while (n--) {
		memset(A, 0, sizeof(A));
		cntp = 0; cntt = 0, other = 0;
		scanf("%s", str);
		int len = strlen(str);
		for (int i = 0; i < len; i++) {
			A[i + 1] = A[i];
			if (str[i] == 'P') {
				cntp++;
				pPos = i;
			}
			else if (str[i] == 'T') {
				cntt++;
				tPos = i;
			}
			else if(str[i]=='A'){
				A[i + 1] = A[i] + 1;
			}
			else {
				other++;
			}
		}
		int x = A[pPos]; // 求出左边区间A的个数
		int y = A[tPos] - A[pPos]; // 求出中间区间A的个数
		int z = A[len] - A[tPos]; //求出右区间A的个数

		if (cntp != 1 || cntt != 1 || other > 0 || y == 0) {
			printf("NO\n");
			continue;
		}

		if ((y == 1 && x == z) || //符合条件2
			x == z - (y - 1)*x) { // 符合条件3化成条件2
			printf("YES\n");
		}
		else {
			printf("NO\n");
		}
	}
	return 0;
}