[이것이 코딩 테스트다 with Python] 이진탐색

교재 : 이것이 코딩 테스트다 with 파이썬
CHAPTER 7 이진 탐색
실전문제 7-1 이진 탐색 189

 

문제

주어진 입력값에 대해서 이진탐색 진행해 인덱스를 출력하자

 

 

입력 조건

숫자의 갯수와 찾아야 할 숫자를 입력받고 다음줄에서 숫자 배열을 입력받는다

 

 

출력 조건

찾은 곳은 인덱스를 출력하고 없다면 원소가 존재하지 않는다는 문구를 출력한다

입력 예시 1
10 7
1 3 5 7 9 11 13 15 17 19

출력 예시 1
4

 

입력 예시 2
10 7
1 3 5 6 9 11 13 15 17 19

출력 예시 2
원소가 존재하지 않습니다.

 

---

✔ Solution - 재귀함수

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)

 

 

✔ Solution - 반복문

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)