PAT乙级-1003 我要通过!
“答案正确”是自动判题系统给出的最令人欢喜的回复。本题属于 PAT 的“答案正确”大派送 —— 只要读入的字符串满足下列条件,系统就输出“答案正确”,否则输出“答案错误”。
得到“答案正确”的条件是:
- 字符串中必须仅有
P
、A
、T
这三种字符,不可以包含其它字符; - 任意形如
xPATx
的字符串都可以获得“答案正确”,其中x
或者是空字符串,或者是仅由字母A
组成的字符串; - 如果
aPbTc
是正确的,那么aPbATca
也是正确的,其中a
、b
、c
均或者是空字符串,或者是仅由字母A
组成的字符串。
现在就请你为 PAT 写一个自动裁判程序,判定哪些字符串是可以获得“答案正确”的。
输入格式:
每个测试输入包含 1 个测试用例。第 1 行给出一个正整数 n (≤10),是需要检测的字符串个数。接下来每个字符串占一行,字符串长度不超过 100,且不包含空格。
输出格式:
每个字符串的检测结果占一行,如果该字符串可以获得“答案正确”,则输出 YES
,否则输出 NO
。
输入样例:
10
PAT
PAAT
AAPATAA
AAPAATAAAA
xPATx
PT
Whatever
APAAATAA
APT
APATTAA
结尾无空行
输出样例:
YES
YES
YES
YES
NO
NO
NO
NO
NO
NO
结尾无空行
做这个题目时,没读懂题目所给的条件,百度了一下
参考 https://pintia.cn/problem-sets/994805260223102976/problems/994805323154440192 得到 白话文 条件
条件解析:
1.字符串中必须有’P’,‘T’,‘A’;
2.‘P’必须在‘T’之前,A穿插其中;
3.P前面的A的个数 乘以 P和T之间的A的个数 等于 T后面的A的个数
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner sc = new Scanner(System.in);
int n = sc.nextInt();
String str = "";
for(int i = 0; i < n; i++){
str = sc.next();
System.out.println(test(str));
}
}
public static String test(String str){
char c[] = str.toCharArray();
int len = c.length;
Map map = new LinkedHashMap();
int sum = 0;
int n = 0;
for(int i = 0; i < len; i++){
if(c[i] == 'A' || c[i] == 'P' || c[i] == 'T'){
//当n==0时,c[i] == 'A' 说明这时 'P'之前的 'A', 将其累加
if(c[i] == 'A' && n == 0) sum ++;
//当n==0时,c[i] == 'P' 说明这是第一个'P' 将'P'送入map中, sum表示'P'之前'A'的个数
if(c[i] == 'P' && n == 0){
map.put(c[i],sum);
n++;
sum = 0;
}else //当n!=0时,c[i] == 'P' 说明这不是第一个'P'
if(c[i] == 'P' && n != 0) return "NO";
//当n==1时,c[i] == 'A',说明这时 'T'之前的 'A', 将其累加
if(c[i] == 'A' && n == 1) sum ++;
//当n==1时,c[i] == 'T',说明第一个'T',将其送入map,并且将'T'之前的'A'送入map,sum表示'P' 'T'之前 'A'的个数
if(c[i] == 'T' && n == 1){
map.put(c[i],0);
if(sum != 0)//sum != 0 说明'P' 'T'之前有'A',将'A'送入
map.put('A',sum);
sum = 0;
n++;
}else //当n!=1时,c[i] == 'T' 说明这不是第一个'T'
if(c[i] == 'T' && n != 1)return "NO";
//统计 'T'后面的 'A'并更新'T'后面'A'的个数
if(c[i] == 'A' && n == 2) map.put('T',(int)map.get('T') + 1);
}else return "NO";
}
if(map.size() == 3 )
if((int)map.get('P') * (int)map.get('A') == (int)map.get('T')) return "YES";
return "NO";
}
}