[이분탐색] 난이도1.5, 이취코 197p '부품 찾기' (Python)

<문제>

동빈이네 전자 매장에는 부품이 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를 출력한다.

---

<문제 풀이>

1. 탐색해야 할 값이 백만 개로 값이 크기 때문에 이분 탐색을 활용
2. 이분탐색을 위해 배열을 정렬시킴

<코드>

'''입력 예시
5
8 3 7 9 2
3 
5 7 9
'''
import sys
input_data = sys.stdin.readline().rstrip()
m = int(input_data.split()[6])
n = int(input_data.split()[0])
m_array = sorted(list(map(int,input_data.split()[7:10])))   #m_array.sort()도 가능
n_array = sorted(list(map(int,input_data.split()[1:6])))

def solution(n_array,target,start,end):
    mid = (start+end)//2

    if start>end:
        return print('no',end = ' ')
    if i==n_array[mid]:
        return print('yes',end = ' ')
    elif i < n_array[mid]:
        solution(n_array,m_array,start,mid-1)
    elif i > n_array[mid]:
        solution(n_array,m_array,mid+1,end)
    else:
        return print('no',end = ' ')

for i in m_array:
	solution(n_array,i,0,n-1)

 

#계수 정렬

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=' ')

 

#220323
#집합 자료형 이용 -> 단순히 특정한 데이터가 존재하는지 검사할 때에 효과적임

n = int(input())
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=' ')

 

<고쳐야 할 점>

• target을 함수 밖에서 써야 함
• 출력 예시 잘 볼 것(enter, space 등)
• set() 써야 함
• 복습 알고리즘