“答案正確”是自動判題系統給出的最令人歡喜的回覆。本題屬於 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
代碼
// b1003-我要通過!.cpp : 此文件包含 "main" 函數。程序執行將在此處開始並結束。
// PAT APATA AAPATAA AAAPATAAA AAAAPAATAAAAAAAA
// 由題意可知,p前面的A的個數①乘以中間的A的個數②等於後面的A的個數③
#include <iostream>
using namespace std;
bool judge(string str) {
int a1 = 0, a2 = 0, a3 = 0;//統計三個位置a的個數
int p = 0, t = 0;
for (int i = 0; i < str.length(); i++) {
if (p == 0) {
//第一段
if (str[i] == 'A') {
a1++;
}
else if (str[i] == 'P') {
p = 1;
}
else {
return false;
}
}
else {
if (t == 0) {
//中間那段
if (str[i] == 'A') {
a2++;
}
else if (str[i] == 'T') {
t = 1;
}
else {
return false;
}
}
else {
//最後一段
if (str[i] == 'A') {
a3++;
}
else {
return false;
}
}
}
}
if (a2 < 1) {
return false;
}
if (a1 * a2 == a3) {
return true;
}
else {
return false;
}
}
int main()
{
int n;
cin >> n;
for (int i = 0; i < n; i++) {
string str;
cin >> str;
if (judge(str)) {
cout << "YES"<<endl;
}
else {
cout << "NO" << endl;
}
}
return 0;
}