백준 1920번: 수찾기 (JAVA)

문제

N개의 정수 A[1], A[2], …, A[N]이 주어져 있을 때, 이 안에 X라는 정수가 존재하는지 알아내는 프로그램을 작성하시오.

입력

첫째 줄에 자연수 N(1 ≤ N ≤ 100,000)이 주어진다. 다음 줄에는 N개의 정수 A[1], A[2], …, A[N]이 주어진다. 다음 줄에는 M(1 ≤ M ≤ 100,000)이 주어진다. 다음 줄에는 M개의 수들이 주어지는데, 이 수들이 A안에 존재하는지 알아내면 된다. 모든 정수의 범위는 -231 보다 크거나 같고 231보다 작다.

출력

M개의 줄에 답을 출력한다. 존재하면 1을, 존재하지 않으면 0을 출력한다.

 

 

 

---

 

풀이

문제 자체는 매우 평이하며, 어렵지 않다. 다만 다른 분의 좋은 코드가 있어서 이를 기록하기 위해 글을 적는다.

 

코드

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

public class Baekjoon1920_수찾기2 {
	public static void main(String[] args)throws IOException {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		br.readLine();
		StringTokenizer st = new StringTokenizer(br.readLine());
		br.readLine();
		Set<String> set = new HashSet<>();
		StringBuilder answer = new StringBuilder();
		
		while(st.hasMoreTokens()) {
			set.add(st.nextToken());
		}

		st = new StringTokenizer(br.readLine());
		while(st.hasMoreTokens()) {
			answer.append(set.contains(st.nextToken()) ? 1 : 0).append("\n");
		}

		System.out.println(answer);
	}
}