항해 99클럽 코딩테스트 스터디 5기 1일차 TIL (자바 / 미들러)

<오늘의 학습 키워드>

- 해쉬셋

- 이분탐색

<오늘의 문제>

제목 : 백준 / 실버4 / 2776번 암기왕

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

<풀이>

해쉬셋 풀이

package BOJ.자료구조.암기왕_2776.insub2004_251013;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.HashSet;
import java.util.Set;
import java.util.StringTokenizer;

public class Main {
    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int T = Integer.parseInt(br.readLine());
        StringBuilder totalSb = new StringBuilder();

        for (int i = 0; i < T; i++) {
            int N = Integer.parseInt(br.readLine());
            Set<Integer> set = new HashSet<>();

            StringTokenizer st1 = new StringTokenizer(br.readLine());
            for (int j = 0; j < N; j++) {  
                set.add(Integer.valueOf(st1.nextToken()));
            }

            int M = Integer.parseInt(br.readLine());  // M 값을 실제로 저장
            StringTokenizer st2 = new StringTokenizer(br.readLine());

            for (int j = 0; j < M; j++) {  
                if (set.contains(Integer.valueOf(st2.nextToken()))) {
                    totalSb.append("1\n");
                } else {
                    totalSb.append("0\n");
                }
            }
        }
        System.out.print(totalSb);
        br.close();
    }
}

- 수첩1에 해당하는 내용은 set에 저장했다.

- 수첩2의 값을 입력받아서 set에 있는지 확인할 때 존재하면 1, 존재하지 않으면 0을 StringBuilder에 append했다.

- 마지막에 T(테스트케이스)가 모두 종료하면 한번에 StringBuilder 객체 안에 있는 내용을 출력 후 종료했다.

 

이분탐색 풀이

package BOJ.자료구조.암기왕_2776.insub2004_251013;


import java.io.BufferedReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.util.ArrayList;
import java.util.List;
import java.util.StringTokenizer;

public class MainBinarySearch {

    public static void main(String[] args) throws IOException {
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int T = Integer.parseInt(br.readLine());
        StringBuilder sb = new StringBuilder();

        for (int t = 0; t < T; t++) {
            int N = Integer.parseInt(br.readLine());
            StringTokenizer st1 = new StringTokenizer(br.readLine());
            List<Integer> list = new ArrayList<>();

            for (int i = 0; i < N; i++) {
                list.add(Integer.parseInt(st1.nextToken()));
            }
            list.sort(null);

            int M = Integer.parseInt(br.readLine());
            StringTokenizer st2 = new StringTokenizer(br.readLine());
            for (int i = 0; i < M; i++) {
                sb.append(binarySearch(list, Integer.parseInt(st2.nextToken()))).append("\n");
            }
        }
        System.out.print(sb);
        br.close();
    }

    public static int binarySearch(List<Integer> list, int target) {

        int start = 0;
        int end = list.size()-1;
        while (start <= end) {
            int mid = (start+end)/2;
            if (list.get(mid) == target) {
                return 1;
            } else if (list.get(mid) > target) {
                end = mid-1;
            } else if (list.get(mid) < target) {
                start = mid+1;
            }
        }
        return 0;
    }
}

- 이분탐색을 하기 위해서는 정렬이 완료된 자료구조가 필요하기 때문에 list의 sort()함수를 이용해서 정렬했다.

- list와 수첩2에 해당되는 숫자들을 이분탐색을 통해 있으면 1 없으면 0을 StringBuilder 객체에 추가했다.

- 마지막에 T(테스트케이스)가 모두 종료하면 한번에 StringBuilder 객체 안에 있는 내용을 출력 후 종료했다.

<결론, 주의할점>

- 해쉬값을 이용한 탐색은 O(1)의 시간복잡도를 가질 것으로 예상

- 이분탐색을 이용한 탐색은 O(logN)의 시간복잡도를 가질 것으로 예상

88584892가 이분탐색, 88581518 해쉬셋

- 예상대로 해쉬를 이용한 풀이가 좀 더 빨랐다.

  + 이분탐색 풀이에는 list.sort()의 시간도 고려해야되긴 하다.