코딩테스트[파이썬]/알고리즘 문제풀이 입문

[이분탐색] - 이분검색

softmoca__ 2024. 2. 6. 22:37
목차

이분검색

임의의 N개의 숫자가 입력으로 주어집니다.

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

 

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

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

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

 

입력예제 1
8 32
23 87 65 12 57 32 99 81

출력예제 1

3

 

 

 

나의 코드

n,m=map(int,input().split())

arr=list(map(int,input().split()))
arr.sort()
l=0
r=n

while l < r:
    mid=(l+r)//2
    if arr[mid]<m:
        l=mid+1
    elif arr[mid]>m:
        r=mid-1
    elif arr[mid]==m:
        print(mid+1)
        break

정렬 후 이진 탐색

 

정답 코드

import sys
sys.stdin=open("input.txt", "r")
n, m=map(int, input().split())
a=list(map(int, input().split()))
a.sort()
lt=0
rt=n-1
while lt<=rt:
    mid=(lt+rt)//2
    if a[mid]==m:
        print(mid+1)
  
        break
    elif a[mid]>m:
        rt=mid-1
    else:
        lt=mid+1

r의 시작 점이 달라 while 조건에 =가 붙은것 말곤 나의 코드와 똑같다