寻找有序数组元素插入位置2

大家注意观察，上面的示例 1:

输入: [1,3,5,6], 待插入元素为 5 输出: 2

这意味着如果出现重复元素，则靠前插入。

如果插入 5，输出 3 ，靠后插入。

榴芒做法

class Solution(object):
    def searchInsert(self, nums, target):
        left,right = 0,len(nums)-1
        while left<=right:
            mid = (left+right)//2
            if target == nums[mid]:
                while mid<len(nums)-1:
                    if nums[mid]!=nums[mid+1]:
                        break
                    mid+=1
                return mid+1
            elif target < nums[mid]:
                right = mid - 1 
            else:
                left = mid + 1 
        
        return left
so=Solution()
s1=[1,3,5,6]
s2=[1,3,5,5,5,6]
target=5
print(so.searchInsert(s1,target))
print(so.searchInsert(s2,target))

class Solution(object):
    def searchInsert(self, nums, target):
        left,right = 0,len(nums)-1
        while left<=right:
            mid = (left+right)//2
            if target == nums[mid]:
                if mid==len(nums)-1 or nums[mid]!= nums[mid+1]:
                #1.先判断target是否为最后一个,是的话可以直接返回了，因为后面已经没有元素了,同时可以防止第二个条件越界
                #2.判断中点后一个和中点是否一致，不一致也可以直接返回了
                    return mid+1
                else:
                    left=mid+1 #如果 中点后一个和中点相等，则可以从后一个开始继续二分搜索
            elif target < nums[mid]:
                right = mid - 1 
            else:
                left = mid + 1 
        
        return left
so=Solution()
s1=[1,3,5,6]
s2=[1,3,5,5,5,6]
target=5
print(so.searchInsert(s1,target))
print(so.searchInsert(s2,target))