PAT（乙级）——1003 我要通过！

 

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

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

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

1. 任意形如 xPATx 的字符串都可以获得“答案正确”，其中 x 或者是空字符串，或者是仅由字母 A 组成的字符串；\
2. 如果 aPbTc 是正确的，那么 aPbATca 也是正确的，其中 a, b, c 均或者是空字符串，或者是仅由字母 A 组成的字符串。

现在就请你为PAT写一个自动裁判程序，判定哪些字符串是可以获得“答案正确”的。

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

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

输入样例：

8
PAT
PAAT
AAPATAA
AAPAATAAAA
xPATx
PT
Whatever
APAAATAA

输出样例：

YES
YES
YES
YES
NO
NO
NO
NO

一、首要条件字符串中只能有“P","A","T"三种字符，或者是空字符null

后面有两种可以判断正确的类型

1.一种是关于P和T之间对称的,如，APATA, AAPATAA, AAAPATAAA, AAAAPATAAAA,……

这里a=c全是A或者null，b就一个A

2.一种是在此基础上，APATA, APAATAA, APAAATAAA.......

AAPATAA, AAPAATAAAA, AAPAAATAAAAAA......

AAAPATAAA, AAAPAATAAAAAA, AAAPAAATAAAAAAAAA......

可以发现类型2是在类型1的基础上的，

这里c中A的个数=a中A的个数×b中A的个数。

解题思路先判别字符串中是否只有“P","A","T"三种字符，或者是空字符null（P和T有且只有1个，A的个数和位置待判断）

 

#include <stdio.h>
#include <iostream>
#include <string>

using namespace std;

int main(){
	int n;
	cin >> n;
	cin.get();//吸收缓冲区的'\n'
	//scanf("%d", &n);
	string str;
	const string& PAT = "PAT";
	while (n--){
		//getline(cin, str);
		cin >> str;
		//scanf("%s", str);
		//cout << str;

		size_t P = str.find("P");
		size_t A = str.find("A");
		size_t T = str.find("T");

		//cout << "p:" << P << "a:" << A << "t:" << T << endl;

		int size = str.size();

		int ip = 0, ia = 0, it = 0;
		bool flag = true;
		//判断输入的字符串只有P,A,T三种字符和NULL
		if (P == string::npos || A == string::npos || T == string::npos)
		{
			cout << "NO" << endl;//确定有PAT
			//break;
			//printf("NO\n");
		}
		else{
			for (int i = 0; i < size; i++){
				if (str[i] == 'P') ip++;
				if (str[i] == 'A') ia++;
				if (str[i] == 'T') it++;
				if (str[i] != 'P' && str[i] != 'A'&& str[i] != 'T') 
				{
					flag = false;
				}
				
			}

				if (ip!=1 || it!=1||ia<1||T<A||T<P||!flag||(T-P)<=1)//确定有且仅有一个P和一个T，其他字符全是A
				{
					cout << "NO" << endl;
					//break;
					//printf("NO\n");
				}
				else {
					string a, b, c;
					a = str.substr(0, P);
					b = str.substr(P + 1, T - P - 1);
					c = str.substr(T + 1, size);
					int la, lb, lc;
					la = a.length();
					lb = b.length();
					lc = c.length();



					//cout << "a=" << a << "\n" << "b=" << b << "\n" << "c=" << c << "\n";

					if ((size == 3 && P != string::npos&&A != string::npos&&T != string::npos) || (la == lc) || ((la*lb) == lc))//三种不同的类型
					{
						cout << "YES" << endl;
						//printf("YES\n");
						//break;
					}
					else {
						cout << "NO" << endl;
						//printf("NO\n");
						//break;
					}
				}
			
		}
		//cout << "N=" << n << endl;
	}
	system("pause");
	return 0;
}