leetcode-面试题 17.19：消失的两个数字

题目

题目连接
给定一个数组，包含从 1 到 N 所有的整数，但其中缺了两个数字。你能在 O(N) 时间内只用 O(1) 的空间找到它们吗？

以任意顺序返回这两个数字均可。

示例 1:

输入: [1]
输出: [2,3]

示例 2:

输入: [2,3]
输出: [1,4]

解题

方法一：位运算

此题是 只出现过一次的两个数 的改版。

由于nums是1~N的，并且少了两个数，总数为N-2
那么再补上1~N的数，总数为2N-2
对于重复的数，位运算可以消掉，后续就和只出现过一次的两个数一样，来求得两个数

class Solution {
public:
    vector<int> missingTwo(vector<int>& nums) {
        int n=nums.size();
        int m=0;
        for(int num:nums){
            m^=num;
        }
        for(int i=1;i<=n+2;i++){
            m^=i;
        }
        int mask=m&(-m);
        vector<int> res(2);
        for(int num:nums){
            if(num&mask){
                res[0]^=num;
            }else{
                res[1]^=num;
            }
        }
        for(int i=1;i<=n+2;i++){
            if(i&mask){
                res[0]^=i;
            }else{
                res[1]^=i;
            }
        }
        return res;
    }
};

方法二：抽屉原理

数组后补两个-1，使用抽屉原理对数组进行排序，O(n)的时间复杂度，再遍历一遍数组找到两个-1的位置就是答案

class Solution {
public:
    vector<int> missingTwo(vector<int>& nums) {
        vector<int> ans;
        nums.push_back(-1);
        nums.push_back(-1);
        for(int i=0;i<nums.size();i++){
            while(nums[i]!=-1&&nums[i]!=i+1){
                swap(nums[i],nums[nums[i]-1]);
            }
        }
        for(int i=0;i<nums.size();i++) if(nums[i]==-1) ans.push_back(i+1);
        return ans;
    }
};