【Python】LeetCode算法_二分查找

35. 搜索插入位置

给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。假设数组中无重复元素。
如：输入：[1,3,5,6], 2 ，输出：1

def searchInsert(nums, target):
"""搜索插入位置"""
low = 0
high = len(nums)
if target <= nums[low]:
return low
if target > nums[high-1]:
return high
while low < high:
mid = low + (high - low)//2
if target < nums[mid]:
high = mid
elif target > nums[mid]:
low = mid +1
else:
return mid
return low

nums_1 = [1,3,5,6]
print searchInsert(nums_1, 2)