mirror of
https://github.com/krahets/hello-algo.git
synced 2026-11-02 11:15:19 +08:00
Yudong Jin
7257641e0e
* Retranslate the complete Japanese online edition from Chinese * Fit Japanese homepage labels and localize heading permalinks * Point Japanese contribution instructions to the Japanese preview * Replace locale conditionals with multilingual UI tables
50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
"""
|
|
File: binary_search_edge.py
|
|
Created Time: 2023-08-04
|
|
Author: krahets ([email protected])
|
|
"""
|
|
|
|
import sys
|
|
from pathlib import Path
|
|
|
|
sys.path.append(str(Path(__file__).parent.parent))
|
|
from binary_search_insertion import binary_search_insertion
|
|
|
|
|
|
def binary_search_left_edge(nums: list[int], target: int) -> int:
|
|
"""二分探索で最も左にある target を探す"""
|
|
# target の挿入位置を探すことに等しい
|
|
i = binary_search_insertion(nums, target)
|
|
# target が見つからなかったので、-1 を返す
|
|
if i == len(nums) or nums[i] != target:
|
|
return -1
|
|
# target が見つかったので、インデックス i を返す
|
|
return i
|
|
|
|
|
|
def binary_search_right_edge(nums: list[int], target: int) -> int:
|
|
"""二分探索で最も右にある target を探す"""
|
|
# 最も左にある target + 1 を探す問題に変換する
|
|
i = binary_search_insertion(nums, target + 1)
|
|
# j は最も右にある target を指し、i は target より大きい最初の要素を指す
|
|
j = i - 1
|
|
# target が見つからなかったので、-1 を返す
|
|
if j == -1 or nums[j] != target:
|
|
return -1
|
|
# target が見つかったので、インデックス j を返す
|
|
return j
|
|
|
|
|
|
"""実行用コード"""
|
|
if __name__ == "__main__":
|
|
# 重複要素を含む配列
|
|
nums = [1, 3, 6, 6, 6, 6, 6, 10, 12, 15]
|
|
print(f"\n配列 nums = {nums}")
|
|
|
|
# 二分探索で左境界と右境界を求める
|
|
for target in [6, 7]:
|
|
index = binary_search_left_edge(nums, target)
|
|
print(f"最も左にある要素 {target} のインデックスは {index}")
|
|
index = binary_search_right_edge(nums, target)
|
|
print(f"最も右にある要素 {target} のインデックスは {index}")
|