【easy】532. K-diff Pairs in an Array

最新推荐文章于 2023-07-20 01:16:17 发布

weixin_30262255

最新推荐文章于 2023-07-20 01:16:17 发布

阅读量73

点赞数

原文链接：http://www.cnblogs.com/sherry-yang/p/8444573.html

版权

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

/*
思路：因为存在k=0的情况，并且答案需要是完全不重复的数对，所以
    可以先存下每个数字出现的次数，如果k=0且每个数字出现次数大于1，则count++
    如果k>0,且当前数字+k之后的数字也出现过，那么count++
*/
class Solution {
public:
    int findPairs(vector<int>& nums, int k) {
        int count = 0;
        int len = nums.size();
        unordered_map<int,int> map;
        for (auto num:nums) map[num]++;
        for (auto a:map){
            if (k==0 && a.second>1) count++;
            if (k>0 && map.count(a.first + k)) count++;
        }
        return count;
    }
};