Baekjoon Online Judge 10815번 파이썬

문제 링크
https://www.acmicpc.net/problem/10815

10815번: 숫자 카드

print(*result) : 리스트의 요소만 출력


정답 코드

import sys

N = int(sys.stdin.readline())
nums_card = list(map(int,sys.stdin.readline().split()))
M = int(sys.stdin.readline())
nums_int = list(map(int,sys.stdin.readline().split()))

nums_card.sort() #상근이가 가지고 있는 숫자 카드들을 정렬. 정수들은 위치에 맞추어 결과를 출력해야하므로 정렬하지 않음.
result = [0] * (M)

for i in range(M):
  start = 0 #숫자 카드 리스트 인덱스 최소값
  end = N-1 #숫자 카드 리스트 인덱스 최대값
  while start <= end:
    mid = (start + end) // 2
    if nums_card[mid] == nums_int[i]:
      result[i] = 1
      break
    elif nums_card[mid] < nums_int[i]:
      start = mid + 1
    else:
      end = mid - 1
      
print(*result) #리스트를 요소만 빼서 출력