Leetcode 540.有序数组中的单一元素
1 题目描述(Leetcode题目链接)
给定一个只包含整数的有序数组,每个元素都会出现两次,唯有一个数只会出现一次,找出这个数。
输入: [1,1,2,3,3,4,4,8,8]
输出: 2
输入: [3,3,7,7,10,11,11]
输出: 10
注意: 您的方案应该在 O ( l o g n ) O(log n) O(logn)时间复杂度和 O ( 1 ) O(1) O(1)空间复杂度中运行。
2 题解
既然题目要求复杂度了,那就必须要思考一下二分法怎么解了,如果不限制的话异或就可以。
二分取
m
i
d
mid
mid,如果
m
i
d
mid
mid正好不等于它两侧值,那么皆大欢喜,
m
i
d
mid
mid就是我们要找的这个数,如果不等于我们就要确定要去哪一侧二分了,判断条件就是两边的奇偶性。
- 如果 n u m s [ m i d ] = = n u m s [ m i d + 1 ] nums[mid]==nums[mid+1] nums[mid]==nums[mid+1],如果 m i d mid mid左侧的数的个数是偶数个,那就是说明目标在 m i d mid mid右侧,反之在左侧。
- 如果 n u m s [ m i d ] = = n u m s [ m i d − 1 ] nums[mid]==nums[mid-1] nums[mid]==nums[mid−1],如果 m i d mid mid右侧的数的个数是偶数个,那就是说明目标在 m i d mid mid左侧,反之在右侧。
class Solution:
def singleNonDuplicate(self, nums: List[int]) -> int:
length = len(nums)
i, j = 0, length - 1
while i < j:
mid = (i+j)//2
if nums[mid] == nums[mid+1]:
if mid%2 == 1:
j = mid - 1
else:
i = mid + 2
elif nums[mid] == nums[mid-1]:
if mid%2 == 0:
j = mid - 2
else:
i = mid + 1
else:
return nums[mid]
return nums[i]