Skip to main content

이진탐색_부품 찾기

문제

note

동빈이네 전자 매장에는 부품이 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

나의 풀이

# 이분탐색 (재귀) 구현
def binary_search(arr, target, start, end):
# 탐색 종료
if start > end:
return "no"

mid = (start + end) // 2
# 중앙값이 찾는 값과 일치할 경우 반환
if arr[mid] == target:
return "yes"
# 중앙값이 찾는 값보다 클 경우 중앙값 기준 왼쪽에서 탐색
elif arr[mid] > target:
return binary_search(arr, target, start, mid - 1)
# 중앙값이 찾는 값보다 작을 경우 중앙값 기준 오른쪽에서 탐색
else:
return binary_search(arr, target, mid + 1, end)

n = int(input()) # 재고 개수 입력
stock_list = list(map(int, input().split())) # 재고 리스트 번호 입력

m = int(input()) # 주문 개수 입력
order_list = list(map(int, input().split())) # 주문 리스트 번호 입력

result = []

# 주문 리스트 순회
for i in order_list:
# 재고 리스트에 주문 번호가 존재할 경우 yes를 result에 추가
# 재고 리스트에 주문 번호가 존재하지 않을 경우 no를 result에 추가
result.append(binary_search(stock_list, i, 0, len(stock_list) - 1))

# 출력
for i in result:
print(i, 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())
x = list(map(int, input().split()))

for i in x:
result = binary_search(array, i, 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())
x = list(map(int, input().split()))

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

동빈나 풀이 (SET)

n = int(input())

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

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

for i in x:
if i in array:
print('yes', end=' ')
else:
print('no', end=' ')