刷题日记 448. 找到所有数组中消失的数字

56 阅读1分钟

携手创作,共同成长!这是我参与「掘金日新计划 · 8 月更文挑战」的第11天,点击查看活动详情

一、题目描述:

448. 找到所有数组中消失的数字

给你一个含 n 个整数的数组 nums ,其中 nums[i] 在区间 [1, n] 内。请你找出所有在 [1, n] 范围内但没有出现在 nums 中的数字,并以数组的形式返回结果。

 

示例 1:

输入:nums = [4,3,2,7,8,2,3,1]
输出:[5,6]

示例 2:

输入:nums = [1,1]
输出:[2]

提示:

  • n == nums.length
  • 1 <= n <= 10^5
  • 1 <= nums[i] <= n

进阶:你能在不使用额外空间且时间复杂度为 O(n) 的情况下解决这个问题吗? 你可以假定返回的数组不算在额外空间内。

二、思路分析:

目标是将每个存在的元素i放到i-1下标处,并让不存在的元素下标(i-1)变为0;

  1. 倘若元素不在正确的位置(当前元素与下标不匹配即nums[i-1]!=i)则将当前元素与元素目标位置的元素交换。
  2. 如果目标位置元素已经是正确元素则将当前位置元素设为0;
  3. 如果碰到0和元素正确的位置直接跳过。
    最后遍历数组,将值为0的下标加一后填入容器内。
    例:
    [4,3,2,7,8,2,3,1]-> (位置0元素为4,位置3(4-1)元素为7。交换4,7)
    [7,3,2,4,8,2,3,1]-> (位置0元素为7,位置6元素为3,交换7,3)
    [3,3,2,4,8,2,7,1]-> (位置0元素为3,位置2元素为2,交换3,2)
    [2,3,3,4,8,2,7,1]-> (位置0元素为2,位置1元素为3,交换2,3)
    [3,2,3,4,8,2,7,1]-> (位置0元素为3,位置3元素为3,将当前位置设为0,指针移到下一个位置)
    [0,2,3,4,8,2,7,1]-> (位置1元素为2,正确向下搜索。) .
    .
    .
    [1,2,3,4,0,0,7,8]

三、AC 代码:

class Solution {
public:
    vector<int> findDisappearedNumbers(vector<int>& nums) {
        vector<int> ans;
        for(int i = 0;i<nums.size();++i)
            {
            if(nums[i] == 0 || nums[i] == i+1){
                continue;
            }
            else if(nums[nums[i]-1] != nums[i])
            {
                int temp = nums[i];
                nums[i] = nums[nums[i]-1];
                nums[temp-1] = temp;
                --i;
            }
            else
            {
                nums[i] = 0;
            }
        }
        for(int i=0;i<nums.size();++i)
            {
            if(0==nums[i])
            {
                ans.push_back(i+1);
            }
        }
        return ans;
    }
};