[Python/파이썬] 백준 10815번 숫자 카드

10815번: 숫자 카드

 

코드

n = int(input())
s = set(map(int, input().split()))
m = int(input())
for i in map(int, input().split()):
    if i in s:
        print(1, end=" ")
    else:
        print(0, end=" ")

 

숫자 카드의 수들을 집합에 넣고, 이 수를 갖고 있는지 없는지 찾아서 출력해주면 된다.

 

다른 방법으로는 이분탐색을 이용하는 방법이 있다. 바로 2번째 줄이 나온 숫자 카드들을 오름차순 정렬하고 이분탐색을 이용하여 숫자카드들에 원하는 숫자가 있는지 찾는 것이다.

n = int(input())
cards = sorted(list(map(int, input().split())))


def bs(target):
    s, e = 0, n-1
    while s <= e:
        mid = (s+e)//2
        if cards[mid] == target:
            return 1
        elif cards[mid] < target:
            s = mid+1
        else:
            e = mid-1
    return 0


m = int(input())
for t in map(int, input().split()):
    print(bs(t), end=" ")