취준/백준
[백준][Python][10816][이분탐색] 숫자 카드 2
puff
2020. 2. 18. 09:55
문제 : https://www.acmicpc.net/problem/10816
10816번: 숫자 카드 2
첫째 줄에 상근이가 가지고 있는 숫자 카드의 개수 N(1 ≤ N ≤ 500,000)이가 주어진다. 둘째 줄에는 숫자 카드에 적혀있는 정수가 주어진다. 숫자 카드에 적혀있는 수는 -10,000,000보다 크거나 같고, 10,000,000보다 작거나 같다. 셋째 줄에는 M(1 ≤ M ≤ 500,000)이 주어진다. 넷째 줄에는 상근이가 몇 개 가지고 있는 숫자 카드인지 구해야 할 M개의 정수가 주어지며, 이 수는 공백으로 구분되어져 있다. 이수도 -10,00
www.acmicpc.net
이분 탐색 문제다.
list.index()는 O(N) 이므로, count를 사용할 경우 O(N^2)가 되어 시간 초과가 된다.
https://wiki.python.org/moin/TimeComplexity
wiki.python.org
따라서, 이분탐색을 적용해야 한다.
이 문제는 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())