给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
示例 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/search-insert-position
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路1
内置功能实现
class Solution(object):
def searchInsert(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: int
"""
nums.append(target)
nums.sort()
return nums.index(target)
执行结果:
通过
显示详情
执行用时:12 ms, 在所有 Python 提交中击败了98.02%的用户
内存消耗:13 MB, 在所有 Python 提交中击败了7.14%的用户
思路2
逐个遍历
class Solution(object):
def searchInsert(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: int
"""
if target>nums[-1]:
return len(nums)
if target<nums[0]:
return 0
for i in range(len(nums)):
if nums[i]>=target:
return i
执行结果:
通过
显示详情
执行用时:16 ms, 在所有 Python 提交中击败了89.43%的用户
内存消耗:12.9 MB, 在所有 Python 提交中击败了7.14%的用户
思路3
二分法
class Solution(object):
def searchInsert(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: int
"""
low=0
high=len(nums)-1
while low<=high:
mid=(low+high)//2
if nums[mid]==target:
return mid
elif nums[mid]>target:
high=mid-1
elif nums[mid]<target:
low=mid+1
return low
执行结果:
通过
显示详情
执行用时:16 ms, 在所有 Python 提交中击败了89.43%的用户
内存消耗:13.1 MB, 在所有 Python 提交中击败了7.14%的用户