【LeetCode】645. Set Mismatch

最新推荐文章于 2020-07-23 11:21:45 发布

原创最新推荐文章于 2020-07-23 11:21:45 发布 · 561 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

LEETCODE 专栏收录该内容

155 篇文章

订阅专栏

本文介绍了一种高效算法，用于在一个包含从1到n的整数序列中找到重复及缺失的数字。通过两次遍历数组，首先标记已出现的数字，并找出重复项；接着检查未被标记为负数的位置，确定缺失的数字。此方法避免了额外空间需求。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

The set S originally contains numbers from 1 to n. But unfortunately, due to the data error, one of the numbers in the set got duplicated to another number in the set, which results in repetition of one number and loss of another number.

Given an array nums representing the data status of this set after the error. Your task is to firstly find the number occurs twice and then find the number that is missing. Return them in the form of an array.

Example 1:
Input: nums = [1,2,2,4]
Output: [2,3]

Note:
The given array size will in the range [2, 10000].
The given array’s numbers won’t have any order.
求1-n中重复的和缺少的一个数。
方法1，很容易想到。

class Solution {
public:
    int a[10001];
    vector<int> findErrorNums(vector<int>& nums) {
        int n=nums.size();
        for(int i=0;i<n;i++)a[nums[i]]++;
        vector<int> ans;
        for(int i=1;i<=n;i++){
            if(a[i]==2)ans.push_back(i);
        }
        for(int i=1;i<=n;i++){
            if(a[i]==0)ans.push_back(i);
        }
        return ans;
    }
};

然而这种需要额外的开辟0(N)的空间，应该进行优化。
方法2：
利用原有的数组，数字出现过就把对应的位置数字变成负数。
最后还是正数的就是缺少的数字。

class Solution {
public:
    vector<int> findErrorNums(vector<int>& nums) {
        int n=nums.size();
        vector<int> ans;
        for(int i=0;i<n;i++){
            int t=abs(nums[i])-1;
            if(nums[t]>0){
                nums[t]*=-1;
            }else{
                ans.push_back(t+1);
            }
        }
        for(int i=0;i<n;i++){
            if(nums[i]>0){
                ans.push_back(i+1);
                break;
            }
        }
        return ans;
    }
};