540. 有序数组中的单一元素
给定一个只包含整数的有序数组,每个元素都会出现两次,唯有一个数只会出现一次,找出这个数。
注意: 您的方案应该在 O(log n)时间复杂度和 O(1)空间复杂度中运行。
示例 1:
输入: [1,1,2,3,3,4,4,8,8]
输出: 2
示例 2:
输入: [3,3,7,7,10,11,11]
输出: 10
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/single-element-in-a-sorted-array
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二分搜索
因为方案应该在 O(log n)时间复杂度(遍历一整个数组为O(N)),所以需要使用查找算法优化,二分搜索最为契合。
但是当mid指针指向非单一元素时,删除(跳过)这一对元素,避免产生新的单一元素。剩下两个子区间中长度为奇数的包含单一元素。
在移动左右指针时,需要考虑五种情况
- mid直接指向单一元素,程序结束,返回值。
- mid指向的元素的同一元素在右边。
- 右边的子区间长度为奇数
- 左边的子区间长度为奇数
- mid指向的元素的同一元素在左边。
- 右边的子区间长度为奇数
- 左边的子区间长度为奇数
class Solution:
def singleNonDuplicate(self, nums: List[int]) -> int:
left = 0
right = len(nums)-1
while left<right:
mid = (left+right)//2
if (mid == 0 or nums[mid] != nums[mid-1]) and nums[mid] != nums[mid+1]:
return nums[mid]
elif nums[mid] == nums[mid+1]:
if (mid-left)%2 == 0:
left = mid + 2
else:
right = mid-1
else:
if (mid-left+1)%2 == 0:
left = mid+1
else:
right = mid - 2
return nums[left]