[백준] 1920 - 수 찾기

 

HashMap이 HashSet보다 결과는 더 빠른 것으로 보인다.

또한 일반 println -> bw.write를 했더니 훨씬 빨라짐을 확인했다.. 앞으로 이걸 쓰자.

import java.io.*;
import java.util.HashMap;
import java.util.StringTokenizer;

public class Main {

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));
        
        HashMap<String, Integer> hm = new HashMap<>();

        int N = Integer.parseInt(br.readLine());

        StringTokenizer st = new StringTokenizer(br.readLine(), " ");
        for(int i=0; i<N; i++) {
            hm.put(st.nextToken(), 0);
        }

        int M = Integer.parseInt(br.readLine());

        st = new StringTokenizer(br.readLine(), " ");
        for(int i=0; i<M; i++) {
            String tmp = st.nextToken();
            if(hm.containsKey(tmp)) {
                bw.write("1\n");
            } else
                bw.write("0\n");
        }
        
        bw.flush();
        bw.close();
    }
}