[BOJ] 백준 10815 숫자 카드 / 이분탐색

이미지를 클릭하면 문제 사이트로 이동합니다.

 

 

💭 문제 이해

사실 이분탐색 문제인데, set 자료구조를 사용해서 in 연산자를 쓰면

원소를 찾는데 O(1) 밖에 안걸린다는 점이 너무나 매력적이다.

 

구현도 엄청 간단하게 하면서 시간도 겁나 빠르다니...

그래서 이번에도 set과 in 연산자를 이용해서 풀었다:)

 

 

 

구현 언어: Python

 

 

import sys

r = sys.stdin.readline


def solve(my_cards: set, num_list: list) -> list:
    result = []
    for num in num_list:
        result.append(1 if num in my_cards else 0)
    return result


n, my_cards = int(r()), set(map(int, r().split()))
m, num_list = int(r()), list(map(int, r().split()))
print(*solve(my_cards, num_list))