[Coding by C++] 我要通过！ (20分)

问题背景

  “答案正确”是自动判题系统给出的最令人欢喜的回复。本题属于 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、条件2好理解，关键在于条件3。

  观察以下描述
    1. 形如 xPATx 的字符串都可以获得“答案正确”
    2. 当 aPbTc 正确时，aPbATca 也正确。
  可得
    aPbTc 需满足 xPATx 条件。

  综上，在最开始时aPbTc仅有以下可能：

//stage 1
PAT    APATA    AAPATAA    AAAPATAAA    AAAAPATAAAA    ······

  随后可推写出 stage 1 对应的 aPbATca

//stage 2
PAAT    APAATAA    AAPAATAAAA    AAAPAATAAAAAA    AAAAPAATAAAAAAAA    ······

  同理由 stage 2 推出 stage 3

//stage 3
PAAAT    APAAATAAA    AAPAAATAAAAAA    AAAPAAATAAAAAAAAA    AAAAPAAATAAAAAAAAAAAA    ······

  不难发现，stage 1 时 A 的个数 a = c，b = 1，随后 b 每增加 1 ，c 都会在原来的基础上加 a ，即 c = c + a。
  可得，a，c为 A 构成的字符串：以P和T为分隔符分割的到三个字符串，满足：第一个字符串长度 x 第二个字符串长度 = 第三个字符串长度

  结合条件1 2 3可得，可通过的字符串应满足以下条件：
    1.有且仅有P、A、T三个字母；
    2.字母 P 一定在字母 T 的左边；
    3. P 和 T 只能出现一次
    4.第一个字符串长度 × 第二个字符串长度 = 第三个字符串长度；
    5.第二个字符串长度不能为 0。

实现代码

#include<iostream>
#include<map>
#include<string>
using namespace std;

int main() {
	int n;
	int indexP;
	int indexT;
	cin >> n;
	cin.get();
	string* ptr = new string[n];
	for (int i = 0; i < n; i++) {
		getline(cin, *(ptr + i));
		map<char, int> mp;
		mp.insert(pair<char, int>('P', 0));
		mp.insert(pair<char, int>('T', 0));
		mp.insert(pair<char, int>('O', 0));   //'O' means other letters
		for (int j = 0; j < (*(ptr + i)).size(); j++) {
			char temp = (*(ptr + i)).at(j);
			if (temp != 'P' && temp != 'A' && temp != 'T') {
				mp['O']++;
			}
			else if (temp == 'P') {
				indexP = j;
				mp['P']++;
			}
			else if (temp == 'T') {
				indexT = j;
				mp['T']++;
			}
		}
		if (mp['O'] != 0) {
			cout << "NO" << endl;
			continue;
		}
		if (mp['P'] != 1 || mp['T'] != 1 || indexP > indexT || indexT - indexP == 1) {
			cout << "NO" << endl;
			continue;
		}
		if (indexP * (indexT - indexP - 1) != (*(ptr + i)).size() - indexT - 1) {
			cout << "NO" << endl;
			continue;
		}
		cout << "YES" << endl;
	}
	delete[] ptr;
}

notes:

Q1：为什么要使用 cin.get() ？
  A：cin >> n 会跳过回车或者空格，使得在 getline 时回车成为第一行字符串。cin.get() 的作用是吸收一个字符，在此处即为 ‘\n’。

  参考博客：
    cin输入数据（cin不丢弃回车符，>>会跳过回车符）
Q2：为什么选择 new 而不是 malloc ？
  A：对于非内部类型来说，malloc 是不能满足要求的。因为 malloc 只是分配堆内存（不会调用构造函数），而 new 是分配堆内存且在此创建一个对象（会调用构造函数）。
  问题的关键在于 string 不是内部类型，而是 C++ 的一个结构，使用 malloc 的话，不会调用构造函数，因此会出现各种奇怪的错误。

  参考博客：
    聊聊malloc和new的区别—当malloc和string相遇时， 容易出错
    最近在使用string类时出现了一个问题，string *p=(string *)malloc(sizeof(string)*10);
Q3：选用 map 的原因？
  A：map 的特点是关键字与值对应，关键字唯一，所以可以用 map 的个数来确定字母的个数。
Q4：char强转string（仅为解题过程中遇到的问题，与本题内容无关）
  A：见示例代码。

//1、构造函数里有个string（size_t,char）
char ch = 'a';
string s(1, ch);
//2、string初始化没char，但是可以利用push_back进行尾插
string s;
s.push_back(ch);
//3、string可以由char*初始化
char ch[2] = { 'a','b' };
string s(ch);

  参考博客：
    c++中char转换为string类型