백준 2776 암기왕(Java)

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

2776번: 암기왕

---

 

해석 및 팁

 

---

 

이 문제를 풀 때 ArrayList나 배열을 사용하면 시간초과가 발생하므로 HashSet을 사용해주어야 합니다. 먼저 처음 주어진 수를 set에 저장한 후 그다음 주어지는 수가 존재하는지 확인하여 존재하면 StringBuilder에 1을 더하고 존재하지 않으면 0을 더하면 됩니다.

 

---

 

Java 코드

 

---

import java.util.Scanner;
import java.util.HashSet;

public class Main {
  public static void main(String[] args) {
    Scanner sc = new Scanner(System.in);
    StringBuilder sb = new StringBuilder();

    int t = sc.nextInt();
  
    for(int i = 0; i < t; i++) {
      
      HashSet<Integer> set = new HashSet<>();
      
      int n = sc.nextInt();
      
      for(int j = 0; j < n; j++) {
        set.add(sc.nextInt());
      }
      
      int m = sc.nextInt();
      
      for(int j = 0; j < m; j++) {
        int num = sc.nextInt();
        if(set.contains(num)) sb.append(1+"\n");
        else sb.append(0+"\n");
      }
      
    }

    System.out.println(sb);
  }
}