645. 错误的集合

题目
O(n)解题思路
代码
优化

toc

题目

集合 s 包含从 1 到 n 的整数。不幸的是，因为数据错误，导致集合里面某一个数字复制了成了集合里面的另外一个数字的值，导致集合 丢失了一个数字 并且 有一个数字重复 。

给定一个数组 nums 代表了集合 S 发生错误后的结果。

请你找出重复出现的整数，再找到丢失的整数，将它们以数组的形式返回。

示例 1：

输入：nums = [1,2,2,4]
输出：[2,3]
示例 2：

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

提示：

2 <= nums.length <= 10^4
1 <= nums[i] <= 10^4

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/set-mismatch
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

O(n)解题思路

根据题意，nums.length取值范围在2到10^4，量不大，nums[i]取值范围也不大，可以考虑利用额外空间配合下标索引换取时间。nums中元素基本是连续的(不是说有序)，额外空间nums.length + 1大小即可
过程简述：

• 定义存储flag的向量，输入值会作为此向量的下标
• 遍历输入，修改下标指向的falg值
• 发现下标指向的flag值已经被修改时，即找到重复的整数
• 继续遍历，直到所有输入完成对flag的标记
• 遍历存储flag的向量，未被标记的下标即为丢失的整数(0处除外)

代码

class Solution {
public:
    vector<int> findErrorNums(vector<int>& nums) {
        int iRepeattion =0, iMissing = 0, ivecFlagSize = nums.size() + 1;
        std::vector<bool> vecFLag(ivecFlagSize, false);
        for(const auto& elem :nums){
            if(vecFLag[elem]){
                iRepeattion = elem;
            }else{
                vecFLag[elem] = true;
            }
        }
        for(int i = 1; i < ivecFlagSize; i++){
            if(!vecFLag[i]){
                iMissing = i;
                break;
            }
        }
        std::vector<int> vecRet = {iRepeattion, iMissing};
        return vecRet;
    }
};

优化

将vecFLag的bool实例化变为int实例化，原因参看<Effective STL>

class Solution {
public:
    vector<int> findErrorNums(vector<int>& nums) {
        int iRepeattion =0, iMissing = 0, ivecFlagSize = nums.size() + 1;
        std::vector<int> vecFLag(ivecFlagSize, 0);
        for(const auto& elem :nums){
            if(1 == vecFLag[elem]){
                iRepeattion = elem;
            }else{
                vecFLag[elem] = 1;
            }
        }
        for(int i = 1; i < ivecFlagSize; i++){
            if(0 == vecFLag[i]){
                iMissing = i;
                break;
            }
        }
        std::vector<int> vecRet = {iRepeattion, iMissing};
        return vecRet;
    }
};

来自为知笔记(Wiz)