刷穿leetcode
给定一个数组,包含从 1 到 N 所有的整数,但其中缺了两个数字。你能在 O(N) 时间内只用 O(1) 的空间找到它们吗?
以任意顺序返回这两个数字均可。
示例 1:
输入: [1]
输出: [2,3]
示例 2:
输入: [2,3]
输出: [1,4]
提示:
nums.length <= 30000
代码:
时间复杂度O(n),空间复杂度O(1)
class Solution {
public:
vector<int> missingTwo(vector<int>& nums) {
int n=nums.size();
int m=n+2;
int x=-1,y=-1;
sort(nums.begin(),nums.end()); //先对数组排好序,利用等差数列的性质来判断缺失哪个数字
if(nums[0]!=1&&nums[n-1]!=m) //特殊判断:如果缺失的正好是第一个数和最后一个数,就直接放回就好了
return {1,m};
if(nums[0]!=1&&nums[n-1]==m){ //如果只缺失第一个数,先把第一个数赋值给x,因为题中说最好用O(1)的空间复杂度,所以只用x,y 2个变量
x=1;
if(n==1) //当给的数组中只有一个元素时,y就是2个数的和除以2
y=((1+m)/2);
}
if(nums[0]==1&&nums[n-1]!=m){ //如果只缺失最后一个数,先把最后一个数赋值给y
y=m;
if(n==1) //当给的数组中只有一个元素时,x就是2个数的和除以2
x=((1+m)/2);
}
for(int i=1;i<n;++i){ //再遍历这个数组
if(nums[i]-nums[i-1]!=1){ //如果出现前后2个数的差不为1,说明他们之间缺少一个数,那这个数就用前后2个数的和再除以2得到
if(x==-1)
x=((nums[i]+nums[i-1])/2);
else if(y==-1)
y=((nums[i]+nums[i-1])/2);
}
if(x!=-1&&y!=-1) //如果x和y都被赋值了,循环就没必要进行了,直接退出就行了
break;
}
return {x,y};
}
};