leetcode 645. 错误的集合(Set Mismatch)

Posted 2020-08-11 zhanzq1

目录

• 题目描述：
• 示例 1:
• 解法：

题目描述：

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

给定一个数组 nums 代表了集合 S 发生错误后的结果。你的任务是首先寻找到重复出现的整数，再找到丢失的整数，将它们以数组的形式返回。

示例 1:

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

注意:

1. 给定数组的长度范围是 [2, 10000]。
2. 给定的数组是无序的。

---

解法：

class Solution {
public:
    vector<int> findErrorNums(vector<int>& nums) {
        long long n = nums.size();
        vector<bool> flag(n+1, false);
        long long sum_val = 0;
        int repeated = 0;
        for(int val : nums){
            sum_val += val;
            if(flag[val]){
                repeated = val;
            }else{
                flag[val] = true;
            }
        }
        int missed = (n+1)*n/2 - (sum_val - repeated);
        return {repeated, missed};
    }
};