【PAT (Basic Level) Practice 】P1003 我要通过！(20分)

【PAT (Basic Level) Practice 】P1003 我要通过！(20分)

题目：

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

---

题目分析：

答案正确判断条件：

1. 字符串只可有’P’，‘A’，‘T’，三种字符
2. 字符串中有且仅有一个连续字符"PAT"，且"PAT"前后拥有**相同数量（可为0）**的字符’A’

分析：字符串有且只有一个’P’与一个’T’ ，我们以’P’，‘T’为分界，记’P’前A的个数为numOfA1，记’P’，‘T’ 间A的个数为numOfA2，记’T’后A的个数为numOfA3。当答案正确时，有numOfA1 == numOfA3并且numOfA2 == 1。若numOfA2>1，我们可以通过条件3来减少numOfA2，观察可得每当numOfA2减少1，numOfA3就会减少numOfA1。直到numOfA2 == 1时判断此时字符串是否正确。

归纳：当numOfA1*numOfA2 == numOfA3时答案正确

代码：

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

//判断是否只有'P' 'A' 'T'，若有其他字符返回0，否则返回1 
int IsPAT(char str[])
{
	int i;
	for(i=0;i<strlen(str);i++)
		if(!(str[i]=='P'||str[i]=='A'||str[i]=='T')) return 0;
	return 1;
}

//是否符合条件2 3 
int Solve(char str[])
{
    //记录'P'，'T'的下标
	int indexOfP, indexOfT;
	int len = strlen(str), i;
	
    //查找'P'，'T'的位置
	for(i=0;i<len;i++)
		if(str[i]=='P')
		{
			indexOfP = i;
			break;
		}
	for(i=len-1;i>=0;i--)
		if(str[i]=='T')
		{
			indexOfT = i;
			break;
		}
	
	if(indexOfT-indexOfP<=1) return 0;//如果P在T后，则不符合
	int numOfA1 = indexOfP;//P前A的个数
	int numOfA2 = indexOfT - indexOfP -1;//P与T中间A的个数
	int numOfA3 = len - indexOfT -1;//T后A的个数
	
	if(numOfA1*numOfA2==numOfA3) return 1;
	return 0;
}

int main()
{
	int n;//储存字符串量
	scanf("%d\n", &n);
	
	char str[101];
	while(n--)
	{
		scanf("%s", str);
		if(IsPAT(str))
		{
			if(Solve(str))
				printf("YES\n");
			else
				printf("NO\n");
		}
		else
		printf("NO\n");
	}
	 
	return 0;
}