PAT-乙1003 我要通过！(20分)C++

题目

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

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

1.字符串中必须仅有 P、 A、 T这三种字符，不可以包含其它字符；
2.任意形如 xPATx 的字符串都可以获得“答案正确”，其中 x 或者是空字符串，或者是仅由字母 A 组成的字符串；
3.如果 aPbTc 是正确的，那么 aPbATca 也是正确的，其中 a、 b、 c 均或者是空字符串，或者是仅由字母 A 组成的字符串。
现在就请你为 PAT 写一个自动裁判程序，判定哪些字符串是可以获得“答案正确”的。

1.1输入格式

每个测试输入包含 1 个测试用例。第 1 行给出一个正整数 n (<10)，是需要检测的字符串个数。接下来每个字符串占一行，字符串长度不超过 100，且不包含空格。

1.2输出格式

每个字符串的检测结果占一行，如果该字符串可以获得“答案正确”，则输出 YES，否则输出 NO。

1.3输入样例

8
PAT
PAAT
AAPATAA
AAPAATAAAA
xPATx
PT
Whatever
APAAATAA

1.4输出样例

YES
YES
YES
YES
NO
NO
NO
NO

思路

条件一：一个字符串中只能出现一个P，若干个A，一个T，别的元素不能出现。
条件二：因为一个字符串中只能出现PAT三个元素，所以形如xPATx的字符串实则为APATA 或者AAPATAA等等。。因此我们只需要根据P前的A元素数量，P与T之间A元素的数量，T后元素的数量，得出规律为：
“P前的A元素数量” * “1” =“T后元素的数量”
条件三：aPbTc 其中a代表了：P前的A元素数量，b代表了：P与T之间A元素的数量，c代表了：T后元素的数量，得出规律为：
“P前的A元素数量” * “P与T之间A元素的数量” =“T后元素的数量”
根据aPbTc的规律，我们可以推出aPbATca的规律为：
“P前的A元素数量” * （“P与T之间A元素的数量”+1 ）=“T后元素的数量”+“P前的A元素数量”

代码实现

#define _CRT_SECURE_NO_WARNINGS
#include<iostream>
#include<stdio.h>
#include<string>
 
using namespace std;
int main()
{
	int a = 0;
	cin >> a;
	string a1;
	int flag = 0;
	int flag1 = 0;
	int flag2 = 0;
	int flag3 = 0;
	for (int i = 0; i < a; i++)
	{
		cin >> a1;
		int num = a1.find('P');
		int num1 = a1.find('T')-num-1;
		int num2 = a1.length()-2 - num1-num;
		for (int j = 0; j < a1.length(); j++)
		{

			if (a1[j] == 'P') {
				flag++;
			}
			else if(a1[j]=='A')
			{
				flag1++;
			}
			else if(a1[j]=='T')
			{
				flag2++;
			}
			else
			{
				flag3++;
			}
		}
		if (flag == 1 && flag1 >= 1 && flag2 == 1 && flag3 == 0) {
			if (num*num1 == num2 || num * num1 + num == num2 + num) {
				printf("YES\n");
				flag = 0;
				flag1 = 0;
				flag2 = 0;
				flag3 = 0;
			}
			else
			{
				printf("NO\n");
				flag = 0;
				flag1 = 0;
				flag2 = 0;
				flag3 = 0;
			}
		}
		else
		{
			printf("NO\n");
			flag = 0;
			flag1 = 0;
			flag2 = 0;
			flag3 = 0;
		}
	
	}
}