92 lines
2.5 KiB
Python
92 lines
2.5 KiB
Python
"""
|
||
LeetCode 解题模板
|
||
|
||
使用方法：
|
||
1. 在 Solution 类中实现你的算法
|
||
2. 在 if __name__ == "__main__" 中添加测试用例
|
||
3. 运行：python solution.py
|
||
"""
|
||
|
||
from typing import List
|
||
|
||
|
||
class Solution:
|
||
"""
|
||
示例：搜索插入位置
|
||
|
||
给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。
|
||
如果目标值不存在于数组中，返回它将会被按顺序插入的位置。
|
||
|
||
请必须使用时间复杂度为 O(log n) 的算法。
|
||
"""
|
||
|
||
def searchInsert(self, nums: List[int], target: int) -> int:
|
||
"""
|
||
搜索插入位置
|
||
|
||
Args:
|
||
nums: 排序数组
|
||
target: 目标值
|
||
|
||
Returns:
|
||
目标值的索引，如果不存在则返回插入位置
|
||
"""
|
||
left, right = 0, len(nums) - 1
|
||
|
||
while left <= right:
|
||
mid = left + (right - left) // 2
|
||
if nums[mid] == target:
|
||
return mid
|
||
elif nums[mid] < target:
|
||
left = mid + 1
|
||
else:
|
||
right = mid - 1
|
||
|
||
return left
|
||
|
||
|
||
def print_test_result(test_name: str, passed: bool):
|
||
"""打印测试结果"""
|
||
status = "✓ 通过" if passed else "✗ 失败"
|
||
print(f"{test_name}: {status}")
|
||
|
||
|
||
def assert_equals(actual: int, expected: int) -> bool:
|
||
"""比较两个整数是否相等"""
|
||
return actual == expected
|
||
|
||
|
||
if __name__ == "__main__":
|
||
solution = Solution()
|
||
|
||
# 测试用例 1: 目标值存在于数组中
|
||
nums1 = [1, 3, 5, 6]
|
||
target1 = 5
|
||
result1 = solution.searchInsert(nums1, target1)
|
||
expected1 = 2
|
||
print_test_result("测试用例 1: 目标值存在", assert_equals(result1, expected1))
|
||
|
||
# 测试用例 2: 目标值不存在，应插入到中间位置
|
||
nums2 = [1, 3, 5, 6]
|
||
target2 = 2
|
||
result2 = solution.searchInsert(nums2, target2)
|
||
expected2 = 1
|
||
print_test_result("测试用例 2: 插入中间位置", assert_equals(result2, expected2))
|
||
|
||
# 测试用例 3: 目标值大于所有元素
|
||
nums3 = [1, 3, 5, 6]
|
||
target3 = 7
|
||
result3 = solution.searchInsert(nums3, target3)
|
||
expected3 = 4
|
||
print_test_result("测试用例 3: 插入末尾", assert_equals(result3, expected3))
|
||
|
||
# 测试用例 4: 目标值小于所有元素
|
||
nums4 = [1, 3, 5, 6]
|
||
target4 = 0
|
||
result4 = solution.searchInsert(nums4, target4)
|
||
expected4 = 0
|
||
print_test_result("测试用例 4: 插入开头", assert_equals(result4, expected4))
|
||
|
||
print("\n所有测试完成！")
|
||
|