面试题 17.19. 消失的两个数字
给定一个数组,包含从 1 到 N 所有的整数,但其中缺了两个数字。你能在 O(N) 时间内只用 O(1) 的空间找到它们吗?
以任意顺序返回这两个数字均可。
示例 1:
输入: [1]
输出: [2,3]
示例 2:
输入: [2,3]
输出: [1,4]
提示:
nums.length <= 30000
解析:
- 将数字存储到原数组对应下标下,例如 1 存到 nums[1] , 2 存到 nums[2] , … ,n 存到 nums[n].
- 原数组少了两个数,因此要添加两个空间,方便代码书写不适用0位置,因此需要额外添加空间,新增位置放入-1。
- 遍历数组,将当前值放到对应下标下。例如,nums[i] = 5,则将nums[5]改为5,将nums[5]原值放到nums[i] 处,继续处理nums[i]。知道遍历到最后为止。
- 遍历结束 ,如果值为-1说明没有该值。
class Solution {
public:
vector<int> missingTwo(vector<int>& nums) {
int n=nums.size();
nums.insert(nums.end(),-1);
nums.insert(nums.end(),-1);
nums.insert(nums.end(),-1);
int cur=nums[0];
for(int i=0;i<n;){
if(nums[i]==i||nums[i]==-1){
i++;
continue;
}
// 如果换过来的值不是-1,需要继续交换这个值
int tmp=nums[i];
nums[i]=nums[nums[i]];
nums[tmp] = tmp;
}
vector<int> res;
for(int i=1;i<n+3;i++){
if(nums[i]==-1)
res.push_back(i);
}
return res;
}
};