【LeetCode】35. 搜索插入位置

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

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

示例 1:

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

示例 2:

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

示例 3:

输入: [1,3,5,6], 7
输出: 4

示例 4:

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

#!/usr/bin/python3
# -*- coding: utf-8 -*-
# @Time: 2018/7/5
# @Author: xfLi
# 解决思路：折半查找算法

def searchInsert(nums, target):
    """
    :type nums: List[int]
    :type target: int
    :rtype: int
    """
    left = 0
    right = len(nums) - 1
    if nums[-1] < target: return len(nums)
    elif nums[0] > target: return 0

    while left <= right:
        mid = (left + right) // 2
        if nums[mid] == target: return mid
        elif nums[mid] > target: right = mid - 1
        elif nums[mid] < target: left = mid + 1
    return left

if __name__ == '__main__':
    s_list = [1,3,5,7]
    key = 6
    idx = searchInsert(s_list, key)
    print(idx)