leetcode [540] 有序数组中的单一元素
【转载】
作者:宫水三叶
链接:https://leetcode.cn/problems/single-element-in-a-sorted-array/solutions/1264496/gong-shui-san-xie-er-duan-xing-fen-xi-yu-17nv/
来源:力扣(LeetCode)
方法一:遍历
数据范围为 1 0 5 10 ^ 5 105,最简单的方法是以「步长为 2」的方式进行从前往后的遍历,找到第一个不符合「与后一个数相等」条件的值即是答案。
或是利用单个元素只有一个(其余成对出现),从头到尾异或一遍,最终结果为单一元素值。
#include <vector>
using std::vector;
/*
* @lc app=leetcode.cn id=540 lang=cpp
*
* [540] 有序数组中的单一元素
*/
// @lc code=start
class Solution {
public:
int singleNonDuplicate(vector<int>& nums) {
int n = nums.size();
for(int i = 0; i < n -1 ; i += 2){
if(nums[i] != nums[i + 1]){
return nums[i];
}
}
return nums[n - 1];
}
};
// @lc code=end
时间复杂度:O(n)
空间复杂度:O(1)
方法二:二分
这道题的「二段性」分析需要一点点「脑筋急转弯」。
示例 1:
输入: nums = [1,1,2,3,3,4,4,8,8]
输出: 2
示例 2:
输入: nums = [3,3,7,7,10,11,11]
输出: 10
由于给定数组有序且常规元素总是两两出现,因此如果不考虑“特殊”的单一元素的话,我们有结论:成对元素中的第一个所对应的下标必然是偶数,成对元素中的第二个所对应的下标必然是奇数。
然后再考虑存在单一元素的情况,假如单一元素所在的下标为 x,那么下标 x 之前(左边)的位置仍满足上述结论,而下标 x 之后(右边)的位置由于 x 的插入,导致结论翻转。
存在这样的二段性,指导我们根据当前二分点 mid 的奇偶性进行分情况讨论:
mid 为偶数下标:根据上述结论,正常情况下偶数下标的值会与下一值相同,因此如果满足该条件,可以确保 mid 之前并没有插入单一元素。正常情况下,此时应该更新 l=mid,否则应当让 r=mid−1,但需要注意这样的更新逻辑,会因为更新 r 时否决 mid 而错过答案,我们可以将否决 mid 的动作放到更新 l 的一侧,即需要将更新逻辑修改为 l=mid+1 和 r=mid ;
mid 为奇数下标:同理,根据上述结论,正常情况下奇数下标的值会与上一值相同,因此如果满足该条件,可以确保 mid 之前并没有插入单一元素,相应的更新 l 和 r。
#include <vector>
using std::vector;
/*
* @lc app=leetcode.cn id=540 lang=cpp
*
* [540] 有序数组中的单一元素
*/
// @lc code=start
class Solution {
public:
int singleNonDuplicate(vector<int>& nums) {
int n = nums.size();
int l = 0 ,r = n - 1;
while(l < r){
int mid = l + r >> 1;
if(mid % 2 == 0){
if(mid + 1 < n && nums[mid] == nums[mid + 1]){
l = mid + 1;
}
else{
r = mid;
}
}else{
if(mid - 1 >= 0 && nums[mid - 1] == nums[mid]){
l = mid + 1;
}else{
r = mid;
}
}
}
return nums[r];
}
};
// @lc code=end