boj1920-수찾기

먼지감자·2021년 6월 5일
0

코딩테스트

목록 보기
8/37

문제 :수찾기
주어진 수(list) 안에 특정 숫자가 존재하는지 찾는 문제
단순 이진 탐색으로 품

import sys

def binary_search(start, end, A, target):
    while start <= end:
        mid = (start+end) // 2
        # print('mid: ', mid)
        if A[mid] == target:
            return mid
        if A[mid] > target:
            end = mid - 1
        if A[mid] < target:
            start = mid + 1
    return None 

if __name__ == "__main__":
    input = sys.stdin.readline

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

    start = 0
    end = len(A)-1
    # print('start. end : ', start, end)

    for mm in m:
        # print("mm", mm)
        if binary_search(start, end, A, mm) is not None:
            print('1')
        else:
            print('0')
profile
ML/AI Engineer

0개의 댓글