[BOJ] 1920 수 찾기

문제

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을 출력한다.

예제 입력 1

5
4 1 5 2 3
5
1 3 7 9 5

예제 출력 1

1
1
0
0
1

---

코드

이전 코드

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

public class Main {
    public static void main(String[] args) throws IOException {
	// TODO Auto-generated method stub
	BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        BufferedWriter bw = new BufferedWriter(new OutputStreamWriter(System.out));

        Hashtable<Integer, String> ht = new Hashtable<>();
        String[] A, B;

        int N = Integer.parseInt(br.readLine());
        String s = br.readLine();
        int M = Integer.parseInt(br.readLine());
        String ss = br.readLine();
        br.close();

        A = s.split(" ");
        B = ss.split(" ");
        for (int i = 0; i < N; i++) {
            ht.put(i, A[i]);
        }

        for (int i = 0; i < M; i++) {
            if (ht.containsValue(B[i])) {
                bw.write(1 + "\n");
            } else {
                bw.write(0 + "\n");
            }
        }

        bw.flush();
        bw.close();
    }

}

-> 시간 초과

정답 코드

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

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());
        String ns = br.readLine();
        int m = Integer.parseInt(br.readLine());
        String ms = br.readLine();

        String nArray[] = ns.split(" ");
        String mArray[] = ms.split(" ");

        for (String ss : nArray) {
            hm.put(ss, 0);
        }

        for (String ss : mArray) {
            if (hm.containsKey(ss)) {
                bw.write("1\n");
            } else {
                bw.write("0\n");
            }
        }

        bw.flush();
    }
}

비교해 본 결과, Hashtable vs HashMap, bw.write 부분, close()부분 이런 것들은 처리시간에 큰 영향을 미치지 않았다.
가장 큰 영향을 끼친 것은 '비교할 값이 key값인가 value값인가' 였다.
이전에도 해시를 사용해 풀긴 했지만 value값이 이용했기 때문에 시간초과가 걸린 것 같다.
간단히 포함되어있는지 유무만 파악할 경우 value가 아닌 key를 사용해야겠다.

이분 탐색 코드

package baekjoon;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.Arrays;
import java.util.StringTokenizer;

public class BOJ1920 {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;

        int n = Integer.parseInt(br.readLine());
        int[] firstArr = new int[n];
        st = new StringTokenizer(br.readLine());
        for (int i = 0; i < n; i++) {
            firstArr[i] = Integer.parseInt(st.nextToken());
        }
        Arrays.sort(firstArr);

        int m = Integer.parseInt(br.readLine());
        st = new StringTokenizer(br.readLine());
        int curNum, answer, left, right, mid;
        for (int i = 0; i < m; i++) {
            curNum = Integer.parseInt(st.nextToken());
            answer = 0;
            left = 0;
            right = n - 1;

            while (left <= right) {
                mid = left + (right - left) / 2;
                if (curNum > firstArr[mid]) {
                    left = mid + 1;
                } else if (curNum < firstArr[mid]) {
                    right = mid - 1;
                } else {
                    answer = 1;
                    break;
                }
            }
            System.out.println(answer);
        }
    }
}