[백준] PS/Java [실랜디]

[백준 2776] 암기왕 - JAVA

SH3542 2025. 3. 13. 17:51

https://www.acmicpc.net/problem/2776

 

set contains 판별

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;

class Main {

  public static void main(String[] args) throws IOException {

    BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
    StringBuilder sb = new StringBuilder();

    int T = Integer.parseInt(br.readLine());

    for (int i = 0; i < T; i++) {
      Set<Integer> s = new HashSet<>();

      br.readLine();
      Arrays.stream(br.readLine().split(" ")).mapToInt(Integer::parseInt).forEach(s::add);

      br.readLine();
      Arrays.stream(br.readLine().split(" "))
          .mapToInt(Integer::parseInt)
          .map(n -> s.contains(n) ? 1 : 0)
          .forEach(n -> sb.append(n).append("\n"));
    }

    System.out.println(sb);
  }
}