leedcode35之搜索插入位置

题目描述

给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。

你可以假设数组中无重复元素。

示例 1:

输入: [1,3,5,6], 5
输出: 2

代码：

def binarySearch(nums, target):
    '''

    :param nums:
    :param target:
    :return:
    '''
    left, right = 0, len(nums) - 1
    while left <= right:
        # (right+left)//2存在内存溢出的情况，故采用下面更优！
        mid = left + (right - left) // 2
        if nums[mid] == target:
            return mid
        elif nums[mid] < target:
            left = mid + 1
        else:
            right = mid - 1

    return left


print("----------测试binarySearch()---------")
nums = [1, 2, 4, 6, 8, 9]
index = binarySearch(nums, 2)
print("index=", index)


def binarySearch1(nums, target):
    '''

    :param nums:
    :param target:
    :return:
    '''
    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

    # 此时left==right
    if target <= nums[left]:
        return left
    else:
        return left + 1


print("----------测试binarySearch1()---------")
nums = [1, 3, 5, 6]
index = binarySearch1(nums, 7)
print("index=", index)

输出：

----------测试binarySearch()---------
index= 1
----------测试binarySearch1()---------
index= 4

总结：在二分查找中，最重要的是边界条件判断。在方法二中，判断条件为Left<right;当跳出循环时，left=right，此时元素nums[left]并未加入判断。所以需要在后面再进一步判断targte是否和nums[left]相等，方法中采用的<=,满足则返回left，反之返回left+1