[Algorithm] 부품 찾기

문제

동빈이네 전자 매장에는 부품이 N개 있다. 각 부품은 정수 형태의 고유한 번호가 있다. 어느 날 손님이 M개의 종류의 부품을 대량으로 구매하겠다며 당일 날 견적서를 요청했다. 동빈이는 때를 놓치지 않고 손님이 문의한 부품 M개 종류를 모두 확인해서 견적서를 작성해야 한다. 이때 가게 안에 부품이 모두 있는지 확인하는 프로그램을 작성해보자.
예를 들어 가게의 부품이 총 5개일 때 부품 번호가 다음과 같다고 하자.

입력

• 첫째 줄에 정수 N이 주어진다. (1<=N<=1,000,000)
• 둘째 줄에는 공백으로 구분하여 N개의 정수가 주어진다. 이때 정수는 1보다 크고 1,000,000 이하이다.
• 셋째 줄에는 정수 M이 주어진다. (1<=M<=100,000)
• 넷째 줄에는 공백으로 구분하여 M개의 정수가 주어진다. 이때 정수는 1보다 크고 10억 이하이다.

출력

• 첫째 줄에 공백으로 구분하여 각 부품이 존재하면 yes를, 없으면 no를 출력한다.

입력 예시

5
8 3 7 9 2
3
5 7 9

출력 예시

no yes yes

 

Solution

아래는 틀린 풀이이다. 왜 틀렸는지 생각해보자.

 

n = int(input())
array = list(map(int, input().split()))
array.sort()
m = int(input())
x = list(map(int, input().split()))
def bs(arr,target,start,end):
    if start>end:
       return None
    mid = (start+end)//2
    if arr[mid]==target:
        return mid
    elif arr[mid]>target:
        return bs(arr,target,start,mid-1)
    else :
        return bs(arr,target,mid+1,end)

for i in range(len(x)):
    if bs(array, x[i], 0, len(array) - 1) == x[i]:
        print('yes')
    else:
        print('no')

 

다음이 맞는 풀이이다. 이진 탐색 기법을 이용하였다.

 

n = int(input())
array = list(map(int, input().split()))
array.sort()
m = int(input())
x = list(map(int, input().split()))
def bs(arr,target,start,end):
    if start>end:
       return None
    mid = (start+end)//2
    if arr[mid]==target:
        return mid
    elif arr[mid]>target:
        return bs(arr,target,start,mid-1)
    else :
        return bs(arr,target,mid+1,end)

for i in x:
    result = bs(array, i, 0, n - 1)
    if result != None:
        print('yes', end=' ')
    else:
        print('no', end=' ')