백준[2776] - 암기왕

• 문제: https://www.acmicpc.net/problem/2776
• 분류: 이분탐색, 자료구조
• 난이도: S4
• 소요시간: 10m
• 자아성찰:
  • 문제유형을 이분탐색으로 보고 들어왔는데, 로직 상 Hash 자료구조를 쓰는 방식이 더 빠르고 로직도 간단할 것 같다는 생각이 들었다.
    • 이분탐색 시간복잡도 O(log N)
    • Hash 자료구조 contains 시간 복잡도 O(1)
  • 이분탐색 방식으로도 풀어보고, Hash 자료구조를 통해서도 풀어보았다.
  • 예상대로 이분탐색은 1704ms / Hash 자료구조는 1600ms가 나왔다.
  • 여러 방식으로 풀이가 가능할 때는, 시간적 여유가 된다면 가능한 방식으로 다 풀어보자.

2776번: 암기왕

---

[이분탐색 풀이, 1704ms]

 

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

public class B_2776_bs {

    static int[] note;
    static int N, M;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;
        StringBuilder sb = new StringBuilder();
        int T = Integer.parseInt(br.readLine());
        while (T-- > 0) {
            N = Integer.parseInt(br.readLine());
            note = new int[N];
            st = new StringTokenizer(br.readLine());
            for (int i = 0; i < N; i++) {
                note[i] = Integer.parseInt(st.nextToken());
            }
            Arrays.sort(note);
            M = Integer.parseInt(br.readLine());
            st = new StringTokenizer(br.readLine());
            for (int i = 0; i < M; i++) {
                int num = Integer.parseInt(st.nextToken());
                sb.append(binarySearch(num) ? 1 : 0).append("\n");
            }
        }
        System.out.println(sb);
    }

    public static boolean binarySearch(int target) {
        // 최초 시작 - 끝 설정
        int L = 0;
        int R = N - 1;
        while (L <= R) {
            // 중간값
            int mid = (L + R) / 2;
            // 일치 시 true
            if (note[mid] == target) {
                return true;
            } else if (note[mid] > target) {
                // mid 값이 더 클 경우 mid 기준 앞의 배열 덩어리로 재시도
                R = mid - 1;
            } else {
                // mid 값이 더 작을 경우 mid 기준 뒤의 배열 덩어리로 재시도
                L = mid + 1;
            }
        }
        return false;
    }
}

 

[HashSet 풀이, 1600ms]

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.*;

public class B_2776_set {

    static Set<Integer> note;

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        StringTokenizer st;
        StringBuilder sb = new StringBuilder();
        int T = Integer.parseInt(br.readLine());
        while (T-- > 0) {
            note = new HashSet<>();
            int N = Integer.parseInt(br.readLine());
            st = new StringTokenizer(br.readLine());
            for (int i = 0; i < N; i++) {
                note.add(Integer.parseInt(st.nextToken()));
            }
            int M = Integer.parseInt(br.readLine());
            st = new StringTokenizer(br.readLine());
            for (int i = 0; i < M; i++) {
                int num = Integer.parseInt(st.nextToken());
                sb.append(note.contains(num) ? 1 : 0).append("\n");
            }
        }
        System.out.println(sb);
    }
}