一、数组---数组中的K-diff数对※※※※※

给定一个整数数组和一个整数 k, 你需要在数组里找到不同的 k-diff 数对。这里将 k-diff 数对定义为一个整数对 (i, j), 其中 i 和 j 都是数组中的数字，且两数之差的绝对值是 k.

示例 1:

输入: [3, 1, 4, 1, 5], k = 2
输出: 2
解释: 数组中有两个 2-diff 数对, (1, 3) 和 (3, 5)。
尽管数组中有两个1，但我们只应返回不同的数对的数量。
示例 2:

输入:[1, 2, 3, 4, 5], k = 1
输出: 4
解释: 数组中有四个 1-diff 数对, (1, 2), (2, 3), (3, 4) 和 (4, 5)。
示例 3:

输入: [1, 3, 1, 5, 4], k = 0
输出: 1
解释: 数组中只有一个 0-diff 数对，(1, 1)。
注意:

数对 (i, j) 和数对 (j, i) 被算作同一数对。
数组的长度不超过10,000。
所有输入的整数的范围在 [-1e7, 1e7]。

思路

思路二：利用set和map。用set存放符合k-diff对中较小的那个数，利用set元素唯一的特点，最后直接返回set的大小即可。map用来存放nums数组中每个元素的数量。遍历nums数组，如果在map中存在nums[i]-k，则将nums[i]-k插入到set中；如果在map中存在nums[i]+k，则将nums[k]插入到set中；map[nums[i]]++。最后返回set大小即可。

class Solution {
public:
    int findPairs(vector<int>& nums, int k) {
        if(nums.size()==0 || k<0) return 0;
        set<int> count;
        map<int,int> value;
        for(int i=0;i<nums.size();i++){
            //返回的是被查找元素的个数。map中不存在相同元素key，所以有返回1，无返回0
            if(value.count(nums[i] - k)) count.insert(nums[i]-k);
            if(value.count(nums[i] + k)) count.insert(nums[i]);
            value[nums[i]]++;//map初始默认值为0，也就是value[nums[i]]=0
        }
        return count.size();
    }
};

 

思路一：分为k==0和k>0两种情况。设置计数器count=0，并最后返回count。首先对nums数组排序。

当k==0时，利用map，每个元素与其数量相对应，遍历map，如果其数量大于1，则count++。返回count。
当k>0时，循环数组nums，如果发现后一个元素与前一个元素相同，则continue，直到找到一个不同的数nums[i]，然后进入内循环，从其后查找是否存在比其大k的数nums[j]，如果存在，count++，并且break。如果nums[j]-nums[i]大于k，则直接break。最后返回count。

class Solution {
public:
    int findPairs(vector<int>& nums, int k) {
        if(nums.size()==0) return 0;
        sort(nums.begin(),nums.end());
        int count = 0;
        if(k==0){
            map<int,int> map_nums;
            for(int i=0;i<nums.size();i++){
                map_nums[nums[i]]++;
            }
            for(map<int,int>::iterator iter=map_nums.begin();iter!=map_nums.end();iter++){
                if(iter->second > 1) count+=1;
            }
        }
        else{
            for(int i=0;i<nums.size()-1;i++){
                if(nums[i+1]==nums[i]) continue;
                for(int j=i+1;j<nums.size();j++){
                    if(nums[j] - nums[i] == k){
                        count++;
                        break;
                    }
                    if(nums[j] - nums[i] > k) break;
                }
            }
        }
        return count;
    }
};