교재 : 이것이 코딩 테스트다 with 파이썬
CHAPTER 7 이진 탐색
실전문제 7-1 이진 탐색 189
주어진 입력값에 대해서 이진탐색 진행해 인덱스를 출력하자
숫자의 갯수와 찾아야 할 숫자를 입력받고 다음줄에서 숫자 배열을 입력받는다
찾은 곳은 인덱스를 출력하고 없다면 원소가 존재하지 않는다는 문구를 출력한다
10 7
1 3 5 7 9 11 13 15 17 19
4
10 7
1 3 5 6 9 11 13 15 17 19
원소가 존재하지 않습니다.
def binary_search(arr,left,right,target):
if left > right:
return None # 엉키면 return None
mid = (left+right)//2 # 정수형으로 다루기 위해서 //쓰는 거 잊지 않기!
if target==arr[mid]:
return mid
elif arr[mid]<target:
return binary_search(arr,mid+1,right,target) # mid는 아니니까 그 뒤부터 해주기
elif target<arr[mid]:
return binary_search(arr,left,mid-1,target) # mid는 아니니까 그 앞부터 해주기
n,t= map(int, input().split())
array = list(map(int, input().split()))
array.sort()
print(binary_search(array,0,len(array)-1,t)+1) # 인덱스니까 +1
# 이진 탐색 소스코드 구현 (재귀 함수)
def binary_search(array, target, start, end):
if start > end:
return None
mid = (start + end) // 2
# 찾은 경우 중간점 인덱스 반환
if array[mid] == target:
return mid
# 중간점의 값보다 찾고자 하는 값이 작은 경우 왼쪽 확인
elif array[mid] > target:
return binary_search(array, target, start, mid - 1)
# 중간점의 값보다 찾고자 하는 값이 큰 경우 오른쪽 확인
else:
return binary_search(array, target, mid + 1, end)
# n(원소의 개수)과 target(찾고자 하는 값)을 입력 받기
n, target = list(map(int, input().split()))
# 전체 원소 입력 받기
array = list(map(int, input().split()))
# 이진 탐색 수행 결과 출력
result = binary_search(array, target, 0, n - 1)
if result == None:
print("원소가 존재하지 않습니다.")
else:
print(result + 1)
# 이진 탐색 소스코드 구현 (반복문)
def binary_search(array, target, start, end):
while start <= end:
mid = (start + end) // 2
# 찾은 경우 중간점 인덱스 반환
if array[mid] == target:
return mid
# 중간점의 값보다 찾고자 하는 값이 작은 경우 왼쪽 확인
elif array[mid] > target:
end = mid - 1
# 중간점의 값보다 찾고자 하는 값이 큰 경우 오른쪽 확인
else:
start = mid + 1
return None
# n(원소의 개수)과 target(찾고자 하는 값)을 입력 받기
n, target = list(map(int, input().split()))
# 전체 원소 입력 받기
array = list(map(int, input().split()))
# 이진 탐색 수행 결과 출력
result = binary_search(array, target, 0, n - 1)
if result == None:
print("원소가 존재하지 않습니다.")
else:
print(result + 1)
이진 탐색
- 시간 복잡도 : 평균 O(logN)
이것이 코딩 테스트다 with 파이썬 - 나동빈 저d