[ESTL] lc645. 错误的集合(STL+数学+哈希表)

Posted 2021-07-28 Ypuyu

文章目录

• • 1. 题目来源
  • 2. 题目解析

1. 题目来源

链接：645. 错误的集合

2. 题目解析

非常常见的题目了，用哈希可以直接解决，数学等差数列求和也是完全可以的，桶排序的思路也是可以的。

具体思路请看 【宫水三叶】一题三解：「计数」&「数学」&「桶排序」，三叶姐出品，必属精品！

在此只写我的一个偷懒做法，直接哈希…

---

时间复杂度：$O(n)$
空间复杂度：$O(n)$

---

STL

class Solution {
public:
    vector<int> findErrorNums(vector<int>& nums) {
        int n = nums.size();
        unordered_map<int, int> hmp;
        for (auto& e : nums) hmp[e] ++ ;
        int a, b;
        for (int i = 1; i <= n; i ++ ) 
            if (!hmp[i]) a = i;
            else if (hmp[i] == 2) b = i;

        return {b, a};      
    }
};