[이진 탐색] 실전 문제 <2> 부품 찾기 / 이것이 취업을 위한 코딩테스트다 with 파이썬

부품 찾기

동빈이네 전자 매장에는 부품이 N개 있다. 각 부품은 정수 형태의 고유한 번호가 있다.

어느 날 손님이 M개 종류의 부품을 대량으로 구매하겠다며 당일 날 견적서를 요청했다.

동빈이는 때를 놓치지 않고 손님이 문의한 부품 M개 종류를 모두 확인해서 견적서를 작성해야 한다.

이때 가게 안에 부품이 모두 있는지 확인하는 프로그램을 작성해보자.

 

예를 들어 가게의 부품이 총 5개일 때 부품 번호가 다음과 같다고 하자.

N = 5

[8, 3, 7, 9, 2]

 

손님은 총 3개의 부품이 있는지 확인 요청했는데 부품 번호는 다음과 같다.

M= 3

[5, 7, 9]

 

이때 손님이 요청한 부품 번호의 순서대로 부품을 확인해 부품이 있으면 yes, 없으면 no를 출력한다.

구분은 공백으로 한다.

 

입력 조건

첫째 줄에 정수 N이 주어진다. (1 <= N <= 1,000,000)

둘째 줄에는 공백으로 구분하여 N개의 정수가 주어진다. 이때 정수는 1보다 크고 1,000,000 이하이다.

셋째 줄에는 정수 M이 주어진다. (1 <= M <= 100,000)

넷째 줄에는 공백으로 구분하여 M개의 정수가 주어진다. 이때 정수는 1보다 크고 1,000,000 이하이다.

 

출력 조건

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

 

입력 예시

5
8 3 7 9 2
3
5 7 9

 

출력 예시

no yes yes

 

---

내 코드

import sys

def binary_search(array, target, start, end):
    if start > end:
        return False
    mid = (start + end) // 2
    if array[mid] == target:
        return True
    elif array[mid] > target:
        return binary_search(array, target, start, mid - 1)
    else:
        return binary_search(array, target, mid + 1, end)


n = int(input())
store = list(sys.stdin.readline().split())
m = int(input())
order = list(sys.stdin.readline().split())

store.sort()

for o in order :
    if not binary_search(store, o, 0, n-1):
        print('no', end=' ')
    else:
        print('yes', end=' ')

 

답안 예시 (이진 탐색)

def binary_search(array, target, start, end):
    while start <= end:
        mid = (start + end) // 2
        if array[mid] == target:
            return mid
        elif array[mid] > target:
            end = mid - 1
        else:
            start = mid + 1
    return None

n = int(input())
array = list(map(int, input().split()))
array.sort()

m = int(input())
target = list(map(int, input().split()))

for t in target:
    result = binary_search(array, t, 0, n-1)
    if result != None:
        print('yes', end = ' ')
    else:
        print('no', end = ' ')

 

답안 예시 (계수 정렬)

n = int(input())
array = [0] * 1000001

for i in input().split():
    array[int(i)] = 1

m = int(input())
target = list(map(int, input().split()))

for t in target:
    if array[t] == 1:
        print('yes', end=' ')
    else:
        print('no', end=' ')

# 입력 예시
5
8 3 7 9 2
3
5 7 9


# 모든 원소의 번호를 포함할 수 있는 크기의 리스트를 만들어야함
array = [0, 0 , ... , 0] 
array = [0, 0, 1, 1, 0, 0, 0, 1, 1, 1, 0 , ... , 0 ] # 인덱스 8, 3, 7, 9, 2 값이 1 됨

target = [5, 7, 9]

array[5] == 0 → 'no'
array[7] == 1 → 'yes'
array[9] == 1 → 'yes'
  

 

답안 예시 (집합 자료형)

set() : 집합 자료형을 초기화하는 함수

집합자료형은 단순히 특정한 데이터가 존재하는지 (한 번이라도 등장했는지) 검사할 때에 매우 효과적으로 사용할 수 있다.

n = int(input())
array = set(map(int, input().split()))

m = int(input())
target = list(map(int, input().split()))

for t in target:
    if t in array:
        print('yes', end=' ')
    else:
        print('no', end=' ')

 

출처

나동빈, 『이것이 취업을 위한 코딩테스트다 with 파이썬』, 한빛미디어(주), 2020년