[백준_1920] 수 찾기 - JAVA

문제 출처

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

풀이

• M의 데이터가 10^5이므로, 일반적인 search는 시간초과가 발생한다.
  따라서 Binary Search를 이용해 풀어야한다.

코드

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader bufferedReader = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer stringTokenizer = new StringTokenizer(bufferedReader.readLine());
        int N = Integer.parseInt(stringTokenizer.nextToken());
        int[] A = new int[N];
        stringTokenizer = new StringTokenizer(bufferedReader.readLine());
        for (int i = 0; i < N; i++) {
            A[i] = Integer.parseInt(stringTokenizer.nextToken());
        }
        Arrays.sort(A);
        stringTokenizer = new StringTokenizer(bufferedReader.readLine());
        int M = Integer.parseInt(stringTokenizer.nextToken());
        stringTokenizer = new StringTokenizer(bufferedReader.readLine());
        for (int i = 0; i < M; i++) {
            int X = Integer.parseInt(stringTokenizer.nextToken());
            int result = Arrays.binarySearch(A, X);
            System.out.println(result >= 0 ? 1 : 0);
        }
    }
}

채점 결과