[Sort] 이코테 “부품 찾기” 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보다 크고 1,000,000 이하이다.

출력 조건

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

입력 예시

5
8 3 7 9 2
3
5 7 9

출력 예시

no yes yes

풀이

• 계수 정렬을 사용해 풀 수 있습니다.

예제 코드

import sys

# 부품 개수 및 부품 번호 기록
n = int(sys.stdin.readline().rstrip())
n_list = list(map(int, sys.stdin.readline().rstrip().split()))
m = int(sys.stdin.readline().rstrip())
m_list = list(map(int, sys.stdin.readline().rstrip().split()))

# 가게 부품 번호 기록
idx_list = [0]*1000000
for idx in n_list:
    idx_list[idx] = 1

# 손님이 요청한 부품 확인
for idx in m_list:
    if (idx_list[idx] != 1):
        print("no", end=" ")
    else:
        print("yes", end=" ")

참고

• 이것이 취업을 위한 코딩 테스트다. with python