[파이썬] [이진탐색] 개념 정리

탐색의 개념

- 특정한 데이터를 찾기 위해 데이터를 찾는 과정

 

탐색의 종류

- 순차 탐색과 이진 탐색

 

순차 탐색

- 리스트 안에 있는 특정한 데이터를 찾기 위해 앞에서부터 데이터를 하나씩 확인하는 방법

- 시간복잡도 : O(N^2)

 

이진 탐색

- 정렬되어있는 리스트에서 탐색 범위를 절반씩 좁혀가며 데이터를 탐색하는 방법

- 이진 탐색은 시작점, 끝점, 중간점을 이용하여 탐색 범위를 설정한다.

- 시간복잡도 : O(LogN)

 

 

 

이진탐색의 구현 방법

0. 이진 탐색을 하기 이전 리스트를 정렬한다.

이미 정렬된 10개의 데이터 중에서 값이 4인 원소를 찾는 예시를 살펴보자.

1. 시작점(0), 끝점(9), 중간점(4)을 설정해준다. 이때 4가 중간점의 왼쪽에 위치해 있기 때문에 이후 중간점의 왼쪽에 대해 탐색한다.

 

4는 중간점보다 왼쪽에 위치해있다.

1-2. 이후 중간점의 왼쪽을 탐색할 것이기에 중간점은 끝점을 재설정하는데 사용된다. (끝점 = 중간점 - 1)

 

2. 시작점(0), 끝점(3), 중간점(1)을 설정해준다. 이때 4가 중간점의 오른쪽에 위치해 있기 때문에 이후 중간점의 오른쪽에 대해 탐색한다.

4는 중간점보다 오른쪽에 위치해있다.

2-2. 이후 중간점의 오른쪽을 탐색할 것이기에 중간점은 시작점을 재설정하는데 사용된다. (시작점 = 중간점 + 1)

 

 

3. 시작점(2), 끝점(3), 중간점(2)을 설정해준다. 이때 4가 중간점의 오른쪽에 위치해 있기 때문에 이후 중간점의 오른쪽에 대해 탐색한다.

4는 중간점보다 오른쪽에 위치해있다.

4. 이후 시작점과 끝점, 중간점이 모두 3이되고 찾고자 하는 숫자인 4를 찾은 뒤, 탐색을 종료한다.

 

 

이진 탐색의 구현 1(재귀적 구현)

# 구현 1 : 재귀적 구현
n, target = 10, 7 # 원소의 개수, 찾고자 하는 값
array = [1, 3, 5, 7, 9, 11, 13, 15, 17, 19]
def binary_search(start, end, target):
    if start > end:
        return False
    mid = (end+start)//2
    
    # 중간점의 값보다 찾고자하는 값이 작은 경우 왼쪽 확인
    if array[mid] > target:
        return binary_search(start, mid-1, target)
        
    # 중간점의 값보다 찾고자하는 값이 큰 경우 오른쪽 확인
    elif array[mid] < target:
        return binary_search(mid+1, end, target)

    # 찾은 경우 중간점 인덱스 반환
    elif array[mid] == target:
        return mid

result = binary_search(0, n-1, target)
if result:
    print(result + 1)
else:
    print('원소가 존재하지 않습니다.')

 

이진 탐색의 구현 2(반복문 구현)

# 구현 2 : 반복문 구현
def binary_search2(target, start, end):
    while True:
        if start > end:
            return False
        mid = (start+end) // 2

        # 중간점의 값보다 찾고자하는 값이 작은 경우 왼쪽 확인
        if array[mid] > target:
            end= mid-1 

        # 중간점의 값보다 찾고자하는 값이 큰 경우 오른쪽 확인
        elif array[mid] < target:
            start = mid+1

        # 찾은 경우 중간점 인덱스 반환    
        else:
            return mid

result = binary_search2(target, 0, n-1)
if result:
    print(result + 1)
else:
    print('원소가 존재하지 않습니다.')

 

 

특정 범위에 속하는 데이터 개수 구하기

- bisect 라이브러리를 사용한다.

- bisect_left(array, x) : 정렬된 순서를 유지하면서 배열 array에 x를 삽입할 가장 왼쪽 인덱스를 반환

- bisect_right(array, x) : 정렬된 순서를 유지하면서 배열 array에 x를 삽입할 가장 오른쪽 인덱스를 반환

from bisect import bisect_left, bisect_right

def count_by_range(array, left_value, right_value):
    right_index = bisect_right(array, right_value)
    left_index = bisect_left(array, left_value)
    return right_index - left_index

# 배열 선언
a = [1, 2, 3, 3, 3, 3, 4, 4, 8, 9]

# 값이 4인 데이터 개수 출력
print(count_by_range(a, 4, 4))

# 값이 [-1, 3] 범위에 있는 데이터 개수 출력
print(count_by_range(a, -1, 3))