645. 错误的集合-LeetCode(C++)

645. 错误的集合

题目

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

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

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

提示：

• 2 <= nums.length <= 104
• 1 <= nums[i] <= 104

示例

示例 1：

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

示例 2：

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

题解 - 数组做哈希表

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