找出数组中差值为k去重后的数对

Posted 2020-10-28 lxing722

思路：已知一个含有重复数字的无序数组，还有一个整数k，找出有多少对不重复的数对(i, j)使得i和j的差刚好为k。由于k有可能为0，而只有含有至少两个相同的数字才能形成数对，那么就是说需要统计数组中每个数字的个数。可以建立每个数字和其出现次数之间的映射，然后遍历哈希表中的数字，如果k为0且该数字出现的次数大于1，则结果count自增1；如果k不为0，且用当前数字加上k后得到的新数字也在数组中存在，则结果count自增1，参见代码如下：

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