[백준] 1920번 - 수 찾기 [Java]

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

 

[백준] 1920번 - 수 찾기 [Java]

---

1.  아이디어

 

Set 자료구조로 간단하게 정수가 존재하는지 판단할 수 있다.

---

2. 문제풀이

 

주어진 N개의 정수를 HashSet에 넣은 후 M개의 수가 이 안에 존재하는지 contains 메서드로 확인하는 방식으로 구현했다.

---

3. 코드

 

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

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));
        StringTokenizer st;

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

        Set<Integer> set = new HashSet<>();
        st = new StringTokenizer(br.readLine());
        while (st.hasMoreTokens()) {
            set.add(Integer.parseInt(st.nextToken()));
        }

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

        st = new StringTokenizer(br.readLine());
        while (st.hasMoreTokens()) {
            if (set.contains(Integer.parseInt(st.nextToken()))) {
                bw.write("1\n");
            } else {
                bw.write("0\n");
            }
        }

        bw.flush();
    }
}

---

4. 후기