목차
이분검색
임의의 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 조건에 =가 붙은것 말곤 나의 코드와 똑같다
'코딩테스트[파이썬] > 알고리즘 문제풀이 입문' 카테고리의 다른 글
[이분탐색]- 뮤직비디오(결정알고리즘) (0) | 2024.02.06 |
---|---|
[이분탐색] - 랜선자르기(결정알고리즘) (0) | 2024.02.06 |
[탐색&시물레이션] - 격자판 회문수 (2) | 2024.02.06 |
[탐색 & 시물레이션] - 스도쿠검사 (4) | 2024.02.06 |
[탐색 &시물레이션] - 봉우리 (0) | 2024.02.05 |