題目描述:
“答案正確”是自動判題系統給出的最令人歡喜的回覆。本題屬於 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一個T,中間末尾和開頭可以隨便插入A。
但是必須滿足開頭的A的個數 * 中間的A的個數 = 結尾的A的個數,而且P和T中間不能沒有A。
代碼:
#include <iostream>
#include <map>
#include <string>
using namespace std;
int main()
{
// 記錄p和t的位置
int p = 0, t = 0;
string str;
int num;
while(cin>>num)
{
for(int i = 0 ; i < num ; i ++)
{
cin>>str;
map<char, int> content ;
for(int j = 0 ; j < str.length() ; j ++)
{
if(str[j] == 'P') p = j;
if(str[j] == 'T') t = j;
// 記錄字符串中其他字符個數
content[str[j]]++;
}
// 約束p的個數爲1,t的個數爲1,a的個數任意,只包含三個字母,p和t之間至少有一個字母(t-p != 1)
// a - b - 1 : a和b之間有多少個位置,多 - 1 是代表把a自己減掉
if(content['P'] == 1
&& content['T'] == 1
&& content['A'] != 0
&& content.size() == 3
&& t-p != 1
&& p * (t - p - 1) == str.length() - t - 1)
{
cout<<"YES"<<endl;
} else {
cout<<"NO"<<endl;
}
}
}
}