백준: 10815 숫자 카드

문제

10815번: 숫자 카드

 

문제 풀이

 N개의 카드와 M개의 카드가 있을 때, M개의 카드를 순차적으로 탐색하여 N개의 카드에 해당 카드가 있다면 1, 아닌 경우는 0을 추력하는 문제이다. 이 문제는 로직이나, 알고리즘 적인 것도 아니고 파이썬에서 `in`을 사용할 때, `set` 또는 `dict`를 사용하면 빠르다는 성질만 알고 있으면 쉽게 풀 수 있다.

 

 N개의 카드의 중복인 경우는 `in`에서 시간만 소비하므로 `set`을 이용하여 중복을 제거한다.

 

코드

from sys import stdin


if __name__ == "__main__":
    n = int(stdin.readline())
    n_card = set(map(int, stdin.readline().split()))
    m = int(stdin.readline())
    m_card = list(map(int, stdin.readline().split()))

    for card in m_card:
        if card in n_card:
            print(1, end=' ')
        else:
            print(0, end=' ')