[구름LEVEL] 방 탈출하기 (Java)

구름LEVEL

접근 방식

먼저 주어진 전체 수열을 set으로 저장하고, 그 다음에 주어지는 M개의 수들의 포함여부를 set에서 체크하여 포함할 경우 1을, 아닐 경우 0을 출력하면 되는 문제이다. 

로직 설명 생략

작성 코드

import java.io.*;
import java.util.*;

class Main {
	public static void main(String[] args) throws Exception {
		BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
		int n = Integer.parseInt(br.readLine());
		String[] arr1 = br.readLine().split(" ");
		int m = Integer.parseInt(br.readLine());
		String[] arr2 = br.readLine().split(" ");		
		
		Set<String> set1 = new HashSet<>(Arrays.asList(arr1));
	
		for(String element : arr2){
			if(set1.contains(element)){
				System.out.println("1");
			}else{
				System.out.println("0");
			}
		}
	}
}

Note

- 로직을 따로 설명할 필요가 없을 정도로 매우 간단하게 풀린 문제이지만 정답율이 50%가 안 되길래 기록을 남겨두었다