[BaekJoon] 10815번: 숫자 카드 (Python)

• Reference
• 이런 유형 되게 많은 것 같다..
  • 그냥 문제 푸는 방식이 똑같음
  • 맵 만들어서 $O(1)$로 비교하는 방식

Solution

import collections
import sys
from typing import List


def solution(sang_cards: List[int], cards: List[int]) -> List[int]:
    answer = []

    card_map = collections.defaultdict(int)
    for c in sang_cards:
        card_map[c] = 1

    for c in cards:
        if c in card_map:
            answer.append(1)
            continue

        answer.append(0)

    return answer


N = int(input())  # 무쓸모
sang_cards = list(map(int, sys.stdin.readline().split()))
M = int(input())  # 무쓸모
cards = list(map(int, sys.stdin.readline().split()))

print(*solution(sang_cards, cards))