PTA 1003 我要通过！ (20 分)

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

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

字符串中必须仅有 P、 A、 T这三种字符，不可以包含其它字符；
任意形如 xPATx 的字符串都可以获得“答案正确”，其中 x 或者是空字符串，或者是仅由字母 A 组成的字符串；
如果 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

最初的代码：

#include<iostream>
#include<string>
#include<map>
using namespace std;
int main()
{
	int n;
	int k = 0, j = 0;
	cin >> n;
	string s;
	map<char, int>m;
	for (int i = 0; i < n; i++) {
		cin >> s;
		for (int i = 0; i < s.length(); i++)
		{
			m[s[i]]++;
		}
		for (int i = 0; i < s.length(); i++) {
			if (s[i] != 'P' && s[i] != 'A' && s[i] != 'T') {
				cout << "NO" << endl;
				break;
			}
			if (m['P'] != 1 || m['T'] != 1 || m['A'] == 0) {
				cout << "NO" << endl;
				break;
			}
			k = s.find('P');
			j = s.find('T');
			if (k*(m['A'] - k - (s.length() - j - 1)) == (s.length() - j-1)) {
				cout << "YES" << endl;
				break;
			}
			else {
				cout << "NO" << endl;
				break;
			}
		}
		m.clear();
	}
	return 0;
}

代码复杂不说，而且过程麻烦
有很多地方值得优化：

第一次优化：

#include<iostream>
#include<string>
#include<map>
using namespace std;
int main()
{
	int n,k = 0, j = 0;
	cin >> n;
	string s;
	map<char, int>m;
	for (int i = 0; i < n; i++) {
		cin >> s;
		for (int i = 0; i < s.length(); i++)
		{
			m[s[i]]++;
		}
		for (int i = 0; i < s.length(); i++) {
			if ((m['P'] == 1 || m['T'] == 1) && (s[i] == 'P' || s[i] == 'A' || s[i] == 'T') && m['A']!=0) {
				k = s.find('P');
				j = s.find('T');
				if (k*(m['A'] - k - (s.length() - j - 1)) == (s.length() - j - 1)) {
					cout << "YES" << endl;
					break;
				}
				else {
					cout << "NO" << endl;
					break;
				}
			}
			else
			{
				cout << "NO" << endl;
				break;
			}
		}
		m.clear();
	}
	return 0;
}

优化之处在于：
之前是考虑了很多NO的情况，但是情况太多了难以判断。所以我们从反面考虑，也就是YES的情况。

第二次优化：

#include<iostream>
#include<string>
#include<map>
using namespace std;
int main()
{
	int n,k = 0, h = 0;
	cin >> n;
	string s;
	map<char, int>m;
	for (int i = 0; i < n; i++) {
		cin >> s;
		for (int j = 0; j < s.length(); j++)
		{
			m[s[j]]++;
			if (s[j] == 'P') k = j;
			if (s[j] == 'T')h = j;
			/*k = s.find('P');
			j = s.find('T');*/
		}
		if ((k*(m['A'] - k - (s.length() - h - 1)) == (s.length() - h - 1))
			&& (m['P'] == 1 || m['T'] == 1) && m.size()==3 && m['A'] != 0)
			cout << "YES" << endl;
		else cout << "NO" << endl;
		m.clear();
	}
	return 0;
}

第一次优化中：

cout << "NO" << endl;
					break;

出现次数过多，那么说明所有的YES情况还是很分散，我们可以将所有的条件一次性判断。

if ((k*(m['A'] - k - (s.length() - h - 1)) == (s.length() - h - 1))
			&& (m['P'] == 1 || m['T'] == 1) && m.size()==3 && m['A'] != 0)

易错点

m.clear()

m.clear();

clear是清空，不释放内存。不断对m进行clear，避免上一次结果的影响，如果不就会出现只能正确判断一种情况。

数组越界

if (s[j] == 'P') k = j;
			if (s[j] == 'T')h = j;
			/*k = s.find('P');
			j = s.find('T');*/

不能对K,S直接查找，因为无法判断是否存在，如果那样，就会存在数组越界的情况