这套题可以直接遍历,找到第一个大于target的数并返回其位置即可,但是时间复杂度为
O
(
n
2
)
O(n^2)
O(n2),题目中明确要求时间复杂度为
O
(
l
o
g
n
)
O(logn)
O(logn),考虑二分查找算法,这道题就是标准的二分查找的一个思路,直接套模板即可,注意边界条件的考虑,如果找到了和target一样大的数,则返回其索引即可,因为它的位置就是我们要插入的位置,当left>right的时候,left所在的位置也是我们应该插入的位置
class Solution(object):
def searchInsert(self, nums, target):
"""
:type nums: List[int]
:type target: int
:rtype: int
"""
left = 0
right = len(nums)-1
while left <= right:
mid = (left+right)/2
if nums[mid] == target:
return mid
elif nums[mid] < target:
left = mid+1
else:
right = mid-1
return left