알고리즘/백준

[백준 - SILVER 4] 1920. 수 찾기 - Python3

jsh5408 2021. 10. 6. 00:04

1920. 수 찾기

https://www.acmicpc.net/problem/1920

 

1920번: 수 찾기

첫째 줄에 자연수 N(1 ≤ N ≤ 100,000)이 주어진다. 다음 줄에는 N개의 정수 A[1], A[2], …, A[N]이 주어진다. 다음 줄에는 M(1 ≤ M ≤ 100,000)이 주어진다. 다음 줄에는 M개의 수들이 주어지는데, 이 수들

www.acmicpc.net

 

내 풀이 - 성공

N = int(input())
A = list(map(int, input().split()))
M = int(input())
B = list(map(int, input().split()))

A.sort()

for b in B:
    TF = 0
    l = 0
    r = N-1
    while l <= r:
        m = (l+r) // 2
        if A[m] == b:
            TF = 1
            break
        if A[m] < b:
            l = m+1
        else:
            r = m-1
    if TF:
        print(1)
    else:
        print(0)

 

A 는 정렬해주고 B 의 숫자들마다 이분탐색 돌리기

TF 로 존재하는지 판단해서 존재하면 1 출력, 아니면 0 출력