寻找有序数组元素插入位置2_有序数组怎么判断插入位置-优快云博客

本文链接：https://blog.youkuaiyun.com/ChengSlime/article/details/106955668

本文深入探讨了二分查找算法的实现细节，特别是在处理含有重复元素的有序数组时的特殊情况。通过两个示例，详细解释了如何定位目标元素的位置，并在元素重复时选择靠前或靠后的插入位置。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

大家注意观察，上面的示例 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))