PAT 乙级 1003 我要通过!(20) C++版 1003. 我要通过!(20)

时间限制
400 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
作者
CHEN, Yue

答案正确”是自动判题系统给出的最令人欢喜的回复。本题属于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

成功条件满足的关系:P前面的字符A个数乘以PT之间A的个数等于T后面的A个数即可

 1 // 1003.cpp : 定义控制台应用程序的入口点。
 2 //
 3 
 4 #include "stdafx.h"
 5 #include<iostream>
 6 #include<string>
 7 #include<algorithm>
 8 
 9 using namespace std;
10 
11 int judge(string str);
12 
13 int main()
14 {
15     int N;
16     string str;
17 
18     cin >> N;
19     getchar();
20 
21     for (int i = 0; i < N; ++i)
22     {
23         getline(cin, str);
24 
25         if (judge(str))
26             cout << "YES" << endl;
27         else
28             cout << "NO" << endl;
29     }
30 
31     return 0;
32 }
33 
34 int judge(string str)
35 {
36     int size = str.size();
37     int p,t;//p t分别记录字符串中P和T的下标
38 
39     if (count(&str[0], &str[size], 'P') != 1)//如果不存在P或者P过多
40         return 0;
41 
42     if (count(&str[0], &str[size], 'T') != 1)//如果不存在T或者T过多
43         return 0;
44 
45     for (int i = 0; i < size; ++i)
46     {
47         if (str[i] != 'P' && str[i] != 'A' && str[i] != 'T')//非PAt字符
48             return 0;
49 
50         if (str[i] == 'P')//记录p的下标
51             p = i;
52 
53         if (str[i] == 'T')//记录t的下标
54             t = i;
55     }
56 
57     if (t < p||p+1==t)//如果t在p前面或者pt紧邻
58         return 0;
59 
60     if (p*(t - p - 1) == size - t - 1)//满足关系
61         return 1;
62     else
63         return 0;
64 }