1003. 我要通过！(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

思路：

第一个条件很容易检查；注意第二个条件实际上说的是PAT左右两侧可以有任意多的A，但是两侧A的数量要相同；第三个条件容易产生误导，需要仔细分析。

“如果 aPbTc 是正确的”，注意该条件中有一个如果，则“aPbTc 正确”不满足时，第三个条件无效。我们先考虑aPbTc何时正确。此处a, b, c或是空字符串或是仅有A组成的字符串，而由第一个条件和第二个条件，aPbTc的第一种正确情况是xPATx，即a == c == x，b == A，x的意义同前。考虑第三个条件的后半句“那么 aPbATca 也是正确的”，对于xPATx即是说xPAATxx也是正确的，依次有xPAAATxxx正确，xPAAAATxxxx正确......注意到，P与T之间每增加一个A，T的右面便多了一个x，设P与T之间A的数目为N_A，P之前的字符数为N_l，T之后的字符数为N_r，则有N_r / N_l == N_A，此可以作为判断依据。

代码如下：

// 4477.cpp : Defines the entry point for the console application.
//


#include"iostream"
#include<string>
#include"string.h"
using namespace std;

int isSurelyCorrect(string& src)
{
	int length=src.length () ;
	int countP=0 ,countT=0,countA= 0;
	for (int i = 0;i < src.length();i++)
	{
		if (src[i] == 'P' )
		{
			countP++;
		}
		if (src[i] == 'T')
		{
			countT++;
		}
		if (src[i] == 'A')
		{
			countA++;
		}
		
	}
	if (countP == 1 && countT == 1 && countA >= 1 && (countP + countT + countA) == length)//
		return 1;
	else return 0;
}
int isPAT(string& src)
{
	int flag = 1;
	int pflag = 0;
	for (int i = 0;i < src.length() - 2;i++)
	{
		if (src[i] == 'P')
		{
			if (src[i + 1] == 'A'&&src[i + 2] == 'T')
			{
				pflag = i;
				flag = 0;
				break;
			}
		}
	}
	if (flag == 0&&(2*pflag+5)==src.length ())
		return 1;
	else
		return 0;
}
int finalJudge(string& src)
{
	int count_a = 0;
	int count_mid = 0;
	int count_final = 0;
	int flag_P = 0;
	int flag_T = 0;
	for (int i = 0;i < src.length();i++)
	{
		if (src[i] != 'P')
			count_a++;
		else
			break;
	}
	for (int i = 0;i < src.length();i++)
	{
		if (src[i] == 'P')
		{
			flag_P = i;
			break;
		}		
	}
	for (int i = flag_P + 1;i < src.length()-flag_P-1;i++)
	{
		if (src[i] != 'T')
		{
			count_mid++;
		}
		else
		{
			flag_T = i;
			break;
		}
			
	}
	
	count_final = src.length() - count_a - count_mid - 2;
	if (count_final == (count_mid )*count_a)
		return 1;
	else
		return 0;

}
int main()
{
	int times;
	cin >> times;
	int total = times;
  int  array[10]={0};
	string str;
	int index = 0;
	while (times)
	{
		cin >> str;
		if (isSurelyCorrect(str))
		{
			if (isPAT(str))
			{
				array[index] = 1;
			}else
				if (finalJudge(str))
				{
					array[index] = 1;
				}
				else
					array[index] = 0;
		}
		index++;
		times--;
	}
	for (int i = 0;i < total;i++)
	{
		if (array[i] == 1)
			cout << "YES"<<endl;
		else
			cout << "NO" << endl;
	}
	
	return 0;
}

别人的使用string函数的代码

#include <string>
#include <iostream>
 
using namespace std;
 
int main()
{
    int n;
    cin >> n;
 
    while (n--)
    {
        string s;
        cin >> s;
 
        size_t p = s.find_first_not_of("A");
        if ((p == string::npos) || (s[p] != 'P'))
        {
            cout << "NO" << endl;
            continue;
        }
 
        size_t t = s.find_first_not_of("A", p + 1);
        if ((t == string::npos) || (t == p + 1) || (s[t] != 'T'))
        {
            cout << "NO" << endl;
            continue;
        }
 
        size_t n = s.find_first_not_of("A", t + 1);
        if (n != string::npos)
        {
            cout << "NO" << endl;
            continue;
        }
 
        if ((s.length() - t - 1) == p * (t - p - 1))
        {
            cout << "YES" << endl;
        }
        else
        {
            cout << "NO" << endl;
        }
    }
 
    return 0;
}