코딩 테스트
백준 : 4185번 CD [파이썬, 이분탐색, 실버5]
Squidward
2022. 10. 1. 18:33
문제
상근이와 선영이는 동시에 가지고 있는 CD를 팔려고 한다. CD를 몇 개나 팔 수 있을까?
입력
입력은 여러 개의 테스트 케이스로 이루어져 있다. 각 테스트 케이스의 첫째 줄에는 상근이가 가지고 있는 CD의 수 N, 선영이가 가지고 있는 CD의 수 M이 주어진다. N과 M은 최대 백만이다. 다음 줄부터 N개 줄에는 상근이가 가지고 있는 CD의 번호가 오름차순으로 주어진다. 다음 M개 줄에는 선영이가 가지고 있는 CD의 번호가 오름차순으로 주어진다. CD의 번호는 십억을 넘지 않는 양의 정수이다. 입력의 마지막 줄에는 0 0이 주어진다. 상근이와 선영이가 같은 CD를 여러장 가지고 있는 경우는 없다.
출력
두 사람이 동시에 가지고 있는 CD의 개수를 출력한다.
예제 입력
3 3
1
2
3
1
2
4
0 0
예제 출력
2
[문제풀이]
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 None
n,m = map(int, input().split())
nArray=[]
mArray=[]
count = 0
for i in range(n):
a = int(input())
if a == 0:
break
nArray.append(a)
for i in range(m):
b = int(input())
if b == 0:
break
mArray.append(b)
for i in range(m):
if (binary_search(mArray[i], nArray)!= None):
count+=1
print(count)
728x90