본문 바로가기

알고리즘/이분탐색

백준 - 숫자 카드

본 알고리즘 풀이는 Routine Study에서 진행하고 있습니다.
저를 포함한 구성원이 대부분 초보이므로, 원하시는분은 언제라도 들어오셔도 좋습니다.

문의는 댓글 바람.

문제 출처 : https://www.acmicpc.net/problem/10815

[문제 설명]

카드가 있으면 1을 출력, 없으면 0을 출력.

[접근 방법]

1.딱 봤을 땐 O(NM) 문제. 연산횟수 50만*50만 => 좋지 않은 선택

2.이분탐색 O(NlogN) 문제. 연산횟수 50만*log50만 => 할만해보임

3.해싱을 이용하면 O(N) + O(M). 연산횟수 50만+50만 => 최고

 

[이분탐색]

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

public class Main {
    static Set<Integer> cardSet = new HashSet<>();
    public static void main(String[] args) throws IOException {
        // 행,컬럼순 (위로 가면 -, 아래로 가면 +, 왼쪽으로 가면 -오른쪽으로 가면 +. 1시방향부터 시작.
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());
        StringTokenizer st = new StringTokenizer(br.readLine()," ");

        for (int i = 0; i < N; i++) {
            cardSet.add(Integer.parseInt(st.nextToken()));
        }

        StringBuilder sb = new StringBuilder();
        int M = Integer.parseInt(br.readLine());
        st = new StringTokenizer(br.readLine()," ");
        for (int i = 0; i < M; i++) {
            int cardNo = Integer.parseInt(st.nextToken());
            if (cardSet.contains(cardNo)) {
                sb.append(1);
            } else {
                sb.append(0);
            }
            sb.append(" ");
        }

        System.out.println(sb.deleteCharAt(sb.length()-1).toString());
    }
}

[해쉬셋]

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

public class Main {
    static Set<Integer> cardSet = new HashSet<>();
    public static void main(String[] args) throws IOException {
        // 행,컬럼순 (위로 가면 -, 아래로 가면 +, 왼쪽으로 가면 -오른쪽으로 가면 +. 1시방향부터 시작.
        BufferedReader br = new BufferedReader(new InputStreamReader(System.in));
        int N = Integer.parseInt(br.readLine());
        StringTokenizer st = new StringTokenizer(br.readLine()," ");

        for (int i = 0; i < N; i++) {
            cardSet.add(Integer.parseInt(st.nextToken()));
        }

        StringBuilder sb = new StringBuilder();
        int M = Integer.parseInt(br.readLine());
        st = new StringTokenizer(br.readLine()," ");
        for (int i = 0; i < M; i++) {
            int cardNo = Integer.parseInt(st.nextToken());
            if (cardSet.contains(cardNo)) {
                sb.append(1);
            } else {
                sb.append(0);
            }
            sb.append(" ");
        }

        System.out.println(sb.deleteCharAt(sb.length()-1).toString());
    }
}
728x90
반응형

'알고리즘 > 이분탐색' 카테고리의 다른 글

백준 - N번째 큰 수  (0) 2022.06.24
백준 - 공유기 설치(Kotlin)  (0) 2022.02.15
백준 - 랜선 자르기(Java)  (0) 2022.01.17
백준 - 수 찾기  (0) 2021.12.25
1337. The K Weakest Rows in a Matrix  (0) 2021.11.08