이 문제는 이것이 코딩테스트다. p197페이지 문제이다. 이진탐색을 이용하지 않고도 쉽게 풀 수 있기만 오늘은 이진탐색을 연습하는 날이기에 이진탐색을 이용하여 풀었다.
이진탐색을 구현할 수만 있다면 문제의 난이도는 쉽다.
아래는 나의 코드이다.
import sys
seller_N = int(sys.stdin.readline().rstrip())
seller = list(map(int,sys.stdin.readline().split()))
customer_N=int(sys.stdin.readline().rstrip())
customer=list(map(int,sys.stdin.readline().split()))
seller.sort()
def search(searchList,target):
low=0
high= len(searchList)
while low<high :
mid = (high + low) // 2
if searchList[mid] == target:
return "yes"
elif searchList[mid] < target:
low= mid+1
else:
high= mid-1
return "no"
for i in customer:
print(search(customer,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(가게의 부품 개수) 입력
n = int(input())
# 가게에 있는 전체 부품 번호를 공백을 기준으로 구분하여 입력
array = list(map(int, input().split()))
array.sort() # 이진 탐색을 수행하기 위해 사전에 정렬 수행
# M(손님이 확인 요청한 부품 개수) 입력
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=' ')
'Problem Solving > 이진 탐색' 카테고리의 다른 글
BOJ14425 문자열 집합 (0) | 2024.02.13 |
---|---|
이분 탐색 개념 (0) | 2024.02.13 |
BOJ2343 기타 레슨 (0) | 2023.04.04 |
BOJ1654 랜선 자르기 (0) | 2023.03.20 |
떡볶이 떡 만들기 (0) | 2023.01.09 |