문제 출처
https://www.acmicpc.net/problem/10815
풀이
- 이분탐색을 이용하여 풀 수 있는 문제이다.
- 카운트 정렬로 풀 수도 있겠지만, 이분 탐색으로 분류된 문제이다.
- Arrays.binarySearch()을 통해 쉽게 풀 수 있다.
코드
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[] arr = new int[N];
stringTokenizer = new StringTokenizer(bufferedReader.readLine());
for (int i = 0; i < N; i++) {
arr[i] = Integer.valueOf(stringTokenizer.nextToken());
}
Arrays.sort(arr);
stringTokenizer = new StringTokenizer(bufferedReader.readLine());
int M = Integer.parseInt(stringTokenizer.nextToken());
stringTokenizer = new StringTokenizer(bufferedReader.readLine());
StringBuilder stringBuilder = new StringBuilder();
for (int i = 0; i < M; i++) {
if (Arrays.binarySearch(arr, Integer.parseInt(stringTokenizer.nextToken())) >= 0) {
stringBuilder.append(1).append(" ");
} else {
stringBuilder.append(0).append(" ");
}
}
System.out.println(stringBuilder);
}
}
채점 결과