题目描述

532. 数组中的K-diff数对

思路

自己没做出来，看代码吧。对哈希表的应用不够熟悉。

代码

class Solution {
public:
    int findPairs(vector<int>& nums, int k) {
        int cnt = 0;
        unordered_map<int, int> m;
        for (int num : nums) {
            m[num]++;
        }
        for (auto a : m) {
            if (k==0 && a.second>1) cnt++;
            if (k>0) cnt += m.count(a.first + k);
        }
        return cnt;
    }
};