540. 有序数组中的单一元素
给定一个只包含整数的有序数组,每个元素都会出现两次,唯有一个数只会出现一次,找出这个数。
示例 1:
输入: [1,1,2,3,3,4,4,8,8]
输出: 2
示例 2:
输入: [3,3,7,7,10,11,11]
输出: 10
注意: 您的方案应该在 O(log n)时间复杂度和 O(1)空间复杂度中运行。
思路:
题目要求O(log n)时间复杂度和 O(1)空间复杂度,可以选择二分查找。本题相对于一般的二分查找来说,数组的长度肯定是奇数的,而且每个元素都会出现两次,唯有一个数只会出现一次,因此可以分奇偶讨论,单一元素肯定在奇数的一侧。
步骤如下:
- 当mid为偶数时,mid两边的数字个数为偶数个;当mid为奇数时,mid两边的数字个数为奇数个。
- 当mid为奇数的时候,将mid左移一位。
- 之后比较mid与mid+1的值是否相等,相等的话在mid的右边,不相等的话在mid的左边。
- 循环直到只剩一个元素。
Python:
class Solution:
def singleNonDuplicate(self, nums: List[int]) -> int:
## 左右边界初始化
l, r = 0, len(nums)-1
## 二分查找
while l < r:
mid = (l + r) // 2
if mid % 2 == 1:
mid -= 1
if nums[mid] == nums[mid+1]:
l = mid + 2
else:
r = mid
## 返回单一元素
return nums[l]