leeetcode 面试题 17.19. 消失的两个数字

刷穿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};
    }
};