1003 我要通过!
“答案正确”是自动判题系统给出的最令人欢喜的回复。本题属于 PAT 的“答案正确”大派送 —— 只要读入的字符串满足下列条件,系统就输出“答案正确”,否则输出“答案错误”。
得到“答案正确”的条件是:
1.字符串中必须仅有 P 、 A 、 T \color{FF0000}{P、 A、 T} P、A、T这三种字符,不可以包含其它字符;
2.任意形如 x P A T x \color{FF0000}{xPATx } xPATx的字符串都可以获得“答案正确”,其中 x \color{FF0000}{x} x 或者是空字符串,或者是仅由字母 A \color{FF0000}{A} A 组成的字符串;
3.如果 a P b T c \color{FF0000}{aPbTc} aPbTc 是正确的,那么 a P b A T c a \color{FF0000}{aPbATca} aPbATca 也是正确的,其中 a 、 b 、 c \color{FF0000}{a、 b、 c } a、b、c 均或者是空字符串,或者是仅由字母 A \color{FF0000}{ A } A 组成的字符串。
现在就请你为 PAT 写一个自动裁判程序,判定哪些字符串是可以获得“答案正确”的。
输入格式:
每个测试输入包含 1 个测试用例。第 1 行给出一个正整数 n (≤10),是需要检测的字符串个数。接下来每个字符串占一行,字符串长度不超过 100,且不包含空格。
输出格式:
每个字符串的检测结果占一行,如果该字符串可以获得“答案正确”,则输出 Y E S \color{FF0000}{ YES } YES ,否则输出 N O \color{FF0000}{ NO } NO。
输入样例:
10
PAT
PAAT
AAPATAA
AAPAATAAAA
xPATx
PT
Whatever
APAAATAA
APT
APATTAA
输出样例:
YES
YES
YES
YES
NO
NO
NO
NO
NO
NO
代码实现
import java.util.Scanner;
public class Main {
public static void main(String[] args) {
Scanner scanner = new Scanner(System.in);
// 读入数据组数n
int caseNum = scanner.nextInt();
while (caseNum-- > 0) {
// 读入字符串
String str = scanner.next();
int len = str.length(), ppos = 0, tpos = 0, pcount = 0, tcount = 0, errCount = 0;
// 遍历字符串
for (int i = 0; i < len; i++) {
char ch = str.charAt(i);
if (ch == 'P') {
// 统计P的数量并记录最后一个P的位置
pcount++;
ppos = i;
} else if (ch == 'T') {
// 统计T的数量并记录最后一个T的位置
tcount++;
tpos = i;
} else if (ch != 'A') {
// 统计非PTA字符的数量
errCount++;
}
}
// 判断是否符合题目要求
if (errCount != 0 || tcount != 1 || pcount != 1 || tpos - ppos <= 1) {
System.out.println("NO");
continue;
}
int x = ppos, y = tpos - ppos - 1, z = len - tpos - 1;
// 判断等式是否成立
if (x * y == z) {
System.out.println("YES");
} else {
System.out.println("NO");
}
}
}
}