Python/알고리즘

[Python/알고리즘] 이분 검색

기록하는_사람 2022. 10. 31. 15:08

문제

임의의 N개의 숫자가 입력으로 주어집니다. N개의 수를 오름차순으로 정렬한 다음 N개의 수 중 한 개의 수인 M이 주어지면 이분검색으로 M이 정렬된 상태에서 몇 번째에 있는지 구하는 프로그램을 작성하세요. 단 중복값은 존재하지 않습니다.

 

입력

첫 줄에 한 줄에 자연수 N(3<=N<=1,000,000)과 M이 주어집니다.

두 번째 줄에 N개의 수가 공백을 사이에 두고 주어집니다.

8 32
23 87 65 12 57 32 99 81

 

출력

첫 줄에 정렬 후 M의 값의 위치 번호를 출력한다.

3

 

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

li.sort()

s = 0
e = len(li) - 1

while True:
    mid = (s + e) // 2
    if li[mid] == M:
        print(mid + 1)
        break
    elif li[mid] > M:
        e = mid - 1
    else:
        s = mid + 1