面试题 17.19. 消失的两个数字
给定一个数组,包含从 1 到 N 所有的整数,但其中缺了两个数字。你能在 O(N) 时间内只用 O(1) 的空间找到它们吗?
以任意顺序返回这两个数字均可。
示例 1:
输入: [1]
输出: [2,3]
示例 2:
输入: [2,3]
输出: [1,4]
提示:
nums.length <= 30000
题解
和这个题目LeetCode 41. 缺失的第一个正数–原地哈希一模一样,只是一个变成了两个。
AC代码
class Solution {
public:
vector<int>res;
vector<int> missingTwo(vector<int>& nums)
{
nums.push_back(-1);
nums.push_back(-1);
for(int i=0;i<nums.size();i++)
{
while(i+1!=nums[i]&&nums[i]>0)
{
//cout<<nums[i]<<" "<<nums[nums[i]-1]<<endl;
swap(nums[i],nums[nums[i]-1]);
}
}
for(int i=0;i<nums.size();i++)
{
if(nums[i]-1!=i)
res.push_back(i+1);
}
return res;
}
};