[백준] 1920. 수 찾기 (Python)

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

1920번: 수 찾기

 

✏️ 이분탐색 문제

✏️ 분명 while문이 끝나야 하는데 자꾸 답이 안 나오고 돌아가서 당황했다. 찾았을 경우 answer를 1로 바꿔주고 break를 안 써준 게 원인이었다😅

 

N = int(input())
N_num = list(map(int, input().split()))

M = int(input())
M_num = list(map(int, input().split()))

N_num.sort()

def binary_search(ground, target):
    s = 0
    e = len(ground) - 1
    answer = 0

    while s <= e:
        mid = (s+e) // 2
        if ground[mid] > target:
            e = mid - 1
        elif ground[mid] < target:
            s = mid + 1
        else:
            answer = 1
            break

    return answer

for num in M_num:
    print(binary_search(N_num, num))