[백준][Python][10816][이분탐색] 숫자 카드 2

문제 : https://www.acmicpc.net/problem/10816

10816번: 숫자 카드 2

이분 탐색 문제다.

 

 

list.index()는 O(N) 이므로, count를 사용할 경우 O(N^2)가 되어 시간 초과가 된다.

https://wiki.python.org/moin/TimeComplexity

TimeComplexity - Python Wiki

따라서, 이분탐색을 적용해야 한다.

 

이 문제는 for 문으로 돌면 시간 초과가 뜬다.

따라서, Counter 를 써서 속도를 좀 빠르게 해야 한다. 물론 Counter도 O(N)이지만 내가 구현한 거보단 빠른 듯.

 

이분 탐색으로 찾는 숫자가 있는지를 확인한 후, 있으면 Counter의 답을 ret 배열에 저장한다.

import sys
from collections import Counter


def binary_search(target, data):
    start = 0
    end = len(data) - 1

    while start <= end:
        mid = (start + end) // 2
        if data[mid] == target:
            return mid 
        elif data[mid] < target:
            start = mid + 1
        else:
            end = mid -1

    return -1

n = int(sys.stdin.readline())
a = [int(i) for i in sys.stdin.readline().rstrip().split()]
m = int(sys.stdin.readline())
b = [int(i) for i in sys.stdin.readline().rstrip().split()]

c = Counter(a)
a.sort()
ret = [0 for _ in range(m)]

for i in range(m):
    idx = binary_search(b[i], a)
    if idx == -1:
        continue
    
    ret[i] = c[b[i]]

print(" ".join([str(i) for i in ret]).rstrip())