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

这道题给了我们一个含有重复数字的无序数组,还有一个整数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;
    }
};

***一定要学会unordered_map,auto等,这个很好写哦

posted @ 2018-02-12 14:59  Sherry_Yang  阅读(105)  评论(0编辑  收藏  举报