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

124 阅读1分钟

Offer 驾到,掘友接招!我正在参与2022春招打卡活动,点击查看活动详情

一、题目描述:

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) 的情况下解决这个问题吗? 你可以假定返回的数组不算在额外空间内。

二、思路分析:

把每个数字nums[i]移动到它对应的下标为nums[i]-1的位置,最终不在数组中的数字对应的位置会被重复数字占据。然后扫描整个数组,找到数组中与其下标不能对应的数字即为结果。

第一步相当于基数排序,复杂度为O(n),第二步遍历数组也是O(n),所以总的时间复杂度为O(n),满足题目要求。

三、AC 代码:

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

        vector<int> result;
        for(i=0; i<nums.size(); i++){
            if(nums[i] != i+1)
                result.push_back(i+1);
        }

        return result;
    }
};

范文参考:

ACM金牌题解 | 不额外开辟空间 | 编程熊 - 找到所有数组中消失的数字 - 力扣(LeetCode) (leetcode-cn.com)