mirror of
https://github.com/krahets/hello-algo.git
synced 2026-11-01 18:55:23 +08:00
Yudong Jin
2f561d3d51
* feat(ko): add the complete Korean online edition * Use UTF-8 for Python CI output on Windows * Remove print-book promotion from the Korean site * fix(ko): preserve emphasis before Korean particles
53 lines
2.3 KiB
Python
53 lines
2.3 KiB
Python
"""
|
|
File: binary_search.py
|
|
Created Time: 2022-11-26
|
|
Author: timi ([email protected])
|
|
"""
|
|
|
|
|
|
def binary_search(nums: list[int], target: int) -> int:
|
|
"""이진 탐색(양 끝을 포함하는 닫힌 구간)"""
|
|
# 닫힌 구간 [0, n-1]을 초기화한다. i와 j는 각각 배열의 첫 원소와 마지막 원소를 가리킨다
|
|
i, j = 0, len(nums) - 1
|
|
# 반복하다 탐색 구간이 비면 종료한다(i > j이면 빈 구간이다)
|
|
while i <= j:
|
|
# 이론적으로 Python의 수는 메모리 크기에 따라 무한히 커질 수 있으므로 큰 수의 범위 초과를 고려할 필요가 없다
|
|
m = (i + j) // 2 # 중간 인덱스 m을 계산한다
|
|
if nums[m] < target:
|
|
i = m + 1 # 이 경우 target은 구간 [m+1, j]에 있다
|
|
elif nums[m] > target:
|
|
j = m - 1 # 이 경우 target은 구간 [i, m-1]에 있다
|
|
else:
|
|
return m # 목표 원소를 찾았으므로 해당 인덱스를 반환한다
|
|
return -1 # 목표 원소를 찾지 못했으므로 -1을 반환한다
|
|
|
|
|
|
def binary_search_lcro(nums: list[int], target: int) -> int:
|
|
"""이진 탐색(왼쪽 끝은 포함하고 오른쪽 끝은 제외하는 구간)"""
|
|
# 왼쪽 끝만 포함하는 구간 [0, n)을 초기화한다. i는 배열의 첫 원소를, j는 마지막 원소 바로 다음 위치를 가리킨다
|
|
i, j = 0, len(nums)
|
|
# 반복하다 탐색 구간이 비면 종료한다(i = j이면 빈 구간이다)
|
|
while i < j:
|
|
m = (i + j) // 2 # 중간 인덱스 m을 계산한다
|
|
if nums[m] < target:
|
|
i = m + 1 # 이 경우 target은 구간 [m+1, j)에 있다
|
|
elif nums[m] > target:
|
|
j = m # 이 경우 target은 구간 [i, m)에 있다
|
|
else:
|
|
return m # 목표 원소를 찾았으므로 해당 인덱스를 반환한다
|
|
return -1 # 목표 원소를 찾지 못했으므로 -1을 반환한다
|
|
|
|
|
|
"""실행 코드"""
|
|
if __name__ == "__main__":
|
|
target = 6
|
|
nums = [1, 3, 6, 8, 12, 15, 23, 26, 31, 35]
|
|
|
|
# 이진 탐색(양 끝을 포함하는 닫힌 구간)
|
|
index = binary_search(nums, target)
|
|
print("목표 원소 6의 인덱스 = ", index)
|
|
|
|
# 이진 탐색(왼쪽 끝은 포함하고 오른쪽 끝은 제외하는 구간)
|
|
index = binary_search_lcro(nums, target)
|
|
print("목표 원소 6의 인덱스 = ", index)
|