[BOJ] 10851. 숫자카드 (python)

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

10815번: 숫자 카드

 

문제

숫자 카드는 정수 하나가 적혀져 있는 카드이다. 상근이는 숫자 카드 N개를 가지고 있다. 정수 M개가 주어졌을 때, 이 수가 적혀있는 숫자 카드를 상근이가 가지고 있는지 아닌지를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 상근이가 가지고 있는 숫자 카드의 개수 N(1 ≤ N ≤ 500,000)이 주어진다. 둘째 줄에는 숫자 카드에 적혀있는 정수가 주어진다. 숫자 카드에 적혀있는 수는 -10,000,000보다 크거나 같고, 10,000,000보다 작거나 같다. 두 숫자 카드에 같은 수가 적혀있는 경우는 없다.

셋째 줄에는 M(1 ≤ M ≤ 500,000)이 주어진다. 넷째 줄에는 상근이가 가지고 있는 숫자 카드인지 아닌지를 구해야 할 M개의 정수가 주어지며, 이 수는 공백으로 구분되어져 있다. 이 수도 -10,000,000보다 크거나 같고, 10,000,000보다 작거나 같다

출력

첫째 줄에 입력으로 주어진 M개의 수에 대해서, 각 수가 적힌 숫자 카드를 상근이가 가지고 있으면 1을, 아니면 0을 공백으로 구분해 출력한다.

 

풀이

문제의 핵심은 탐색이다.
카드 목록 중, 현재 숫자(number)가 존재하는지를 확인해야한다.
해보지는 않았지만 number in list 이렇게 하면 시간이 될지는 모르겠다.

그러나 카드 목록의 수가 500000개이므로, 이분탐색을 이용했다.
(이분탐색을 설명하진 않겠다!)
그래도 효율적인 풀이 시간이 나오지는 않아서, 더 좋은 방법이 있다면 고민해보자.

 

코드

import sys
read = sys.stdin.readline

sanggeun_card_N = int(read().strip())
card_list = list(map(int, read().strip().split()))

number_N = int(read().strip())
number_list = list(map(int, read().strip().split()))

is_sanggeun_own = [0] * number_N

card_list.sort()


for i in range(len(number_list)):
    left_index = 0
    right_index = len(card_list) - 1
    while left_index <= right_index:
        mid_index = (left_index + right_index) // 2

        if card_list[mid_index] < number_list[i]:
            left_index = mid_index + 1
        elif card_list[mid_index] > number_list[i]:
            right_index = mid_index - 1
        else:
            is_sanggeun_own[i] = 1
            break
print(*is_sanggeun_own)