코딩테스트 with PYTHON
[python] 코딩테스트 대비 - 이분검색(결정알고리즘)
탱이나라
2022. 1. 14. 21:38
문제)
답안 코드)
이분 탐색은 시간 복잡도 log_2 N을 보장한다.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
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
|
cs |