[PS] 99클럽 코테 스터디 14일차 TIL (숫자카드2)

문제

문제 링크

풀이

HashMap 의 getOrDefault() 함수를 이용해 각 번호가 몇번씩 등장하는지 저장한 후 입력되는 숫자가 해시맵에 몇개 있는지 반환한다.

// 6 3 2 10 10 10 -10 -10 7 3 입력이 주어질때
map = {2=1, 3=2, 6=1, 7=1, -10=2, 10=3}

해시맵의 key 값으로 해당 숫자를 저장하고, value 값으로 등장 횟수를 저장한다.

public class Main {

  static FastReader scan = new FastReader();
  static StringBuilder sb = new StringBuilder();

  public static void main(String[] args) {
    input();
  }

  static void input() {
    int N = scan.nextInt();
    HashMap<Integer, Integer> map = new HashMap<>();
    for (int i = 0; i < N; i++) {
      int key = scan.nextInt();
      map.put(key, map.getOrDefault(key, 0) + 1);
    }
    int M = scan.nextInt();
    for (int i = 0; i < M; i++) {
      sb.append(map.getOrDefault(scan.nextInt(), 0)).append(" ");
    }
    System.out.println(sb);
  }

  static class FastReader {
    BufferedReader br;
    StringTokenizer st;

    public FastReader() {
      br = new BufferedReader(new InputStreamReader(System.in));
    }

    public FastReader(String s) throws FileNotFoundException {
      br = new BufferedReader(new FileReader(new File(s)));
    }

    String next() {
      while (st == null || !st.hasMoreElements()) {
        try {
          st = new StringTokenizer(br.readLine());
        } catch (IOException e) {
          e.printStackTrace();
        }
      }
      return st.nextToken();
    }

    int nextInt() {
      return Integer.parseInt(next());
    }

    long nextLong() {
      return Long.parseLong(next());
    }

    double nextDouble() {
      return Double.parseDouble(next());
    }

    String nextLine() {
      String str = "";
      try {
        str = br.readLine();
      } catch (IOException e) {
        e.printStackTrace();
      }
      return str;
    }
  }

}