mirror of
https://github.com/krahets/hello-algo.git
synced 2026-10-29 11:44:33 +08:00
71a773df64
* docs(ru): correct proofreading issues in prose and code examples * docs(ru): align proofreading edits with Simplified Chinese source * fix(ru): repair generated Zig statements and complete comment translations
53 lines
3.1 KiB
Python
53 lines
3.1 KiB
Python
"""
|
||
File: binary_search.py
|
||
Created Time: 2022-11-26
|
||
Author: timi (xisunyy@163.com)
|
||
"""
|
||
|
||
|
||
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
|
||
|
||
|
||
"""Driver Code"""
|
||
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)
|