【java】【PAT乙】1003 我要通过 (20 分)

博客查看本文

思路和 柳神 的一样，不再多说。
引入哈希进行记录即可。
这里声明一下两个要点：

• 为什么 if 判断里要优先判断 hashmap 的 size
  因为要防止 PT 这种情况出现，虽然 P 和 T 都存在，但是此时 hashmap 中不存在 A ，因此如果不提前判断这个，会在 map.get(‘A’) 时报错，因为 A 根本就没在原字符串中，因此无法形成 A:0 这样的映射关系，这点和柳神的代码不太一样，一定要注意。
  当然还有一种避免该问题的方法就是：不使用 java 自带的 hashmap ，引入一个数组作为 hashmap ，数组初始值均为0，这里可以保证字符串中不存在 A 时，A 的映射值为0。

• 为什么引入了一个字符串数组？
  按照柳神的思路去复现 java 代码的话，仔细观察会发现，每次 yes or no 的判断，当你输入一个字符串以后会立刻显示出来，这与题目要求的先输入指定数量的字符串再显示判断结果的要求不符。因此借一步使用字符串数组存储输入的字符串集合。

import java.util.HashMap;
import java.util.Scanner;

public class Main {
    public static void main(String[] args)   {
        Scanner scanner = new Scanner(System.in);
        int n = scanner.nextInt();
        String[] arr = new String[n];
        scanner.nextLine();
        int p_index = 0, t_index = 0;
        for(int i = 0; i < n; i++){
            String s = scanner.next();
            arr[i] = s;
        }
        for(int i = 0; i < n; i++){
            HashMap<Character, Integer> map = new HashMap<>();
            String s = arr[i];
            for(int j = 0; j < s.length(); j++){
                map.put(s.charAt(j), map.getOrDefault(s.charAt(j), 0)+1);
                if(s.charAt(j)=='P'){
                    p_index = j;
                }
                if(s.charAt(j)=='T'){
                    t_index = j;
                }
            }
            if( map.size() == 3 && map.get('P') == 1 && map.get('A') != 0 && map.get('T') == 1 && t_index-p_index != 1
                    && p_index * (t_index-p_index-1) == s.length()-t_index-1 ){
                System.out.println("YES");
            }
            else {
                System.out.println("NO");
            }

        }

    }
}