面试题 17.19. 消失的两个数字(每日一难phase2--day23)

面试题 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; 
    }
};

你可能感兴趣的:(每日一难,算法,数组)