[이것이 코딩 테스트다 with Python] 부품 찾기 : 이진 탐색

교재 : 이것이 코딩 테스트다 with 파이썬
CHAPTER 7 이진 탐색
실전문제 7-2 부품 찾기 197p

 

문제

동빈이네 전자 매장에는 부품이 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보다 크고 10억 이하이다.

 

출력 조건

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

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

출력 예시
no yes yes

 

 

---

✔ Solution - 이진탐색

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())
store = list(map(int, input().split()))
store.sort()  # 이진 탐색을 위해 정렬

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

for i in customer:
    result = binary_search(store, i, 0, n - 1)
    if result != None:
        print('yes', end=' ')
    else:
        print('no', end=' ')

 

 

✔ Solution 2 - 계수 정렬

# 가게에 있는 물품
n = int(input())   # 가게에 있는 부품의 개수
store = [0] * 1000001  # 가게의 부품 list (0으로 초기화)

# 가게에 있는 부품 check(있는경우 1을 할당)
for i in list(map(int, input().split())):
    store[i] = 1

# 손님이 원하는 물품
m = int(input())  # 손님이 찾는 부품의 개수
customer = list(map(int, input().split()))  # 손님이 원하는 부품 list


# 손님이 찾는 부품과 가게의 부품 비교(있으면 1, 없으면 0)
for i in customer:
    if store[i] == 1:
        print('yes', end=' ')
    else:
        print('no', end=' ')

 

 

✔ Solution 3 - set을 이용한 탐색

n = int(input())
# 가게에 있는 전체 부품 번호를 입력 받아서 집합(Set) 자료형에 기록
store = set(map(int, input().split()))

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

# 손님이 확인 요청한 부품 번호를 하나씩 확인
for i in customer:
    # 해당 부품이 존재하는지 확인
    if i in store:
        print('yes', end=' ')
    else:
        print('no', end=' ')