数组中差为K的对数/K-diff Pairs in an Array(C++)

最新推荐文章于 2022-06-16 23:43:35 发布

TinnCHEN

最新推荐文章于 2022-06-16 23:43:35 发布

阅读量385

点赞数

分类专栏：数据结构

本文链接：https://blog.youkuaiyun.com/TinnCHEN/article/details/104386390

版权

数据结构专栏收录该内容

10 篇文章

订阅专栏

本文介绍了一种解决寻找数组中唯一k-diff数对数量的方法。k-diff数对定义为数组中两个数的绝对差等于k的整数对。通过使用排序和二分搜索算法，文章详细阐述了如何高效地找到所有符合条件的数对。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Problem:
Given an array of integers and an integer k, you need to find the number of unique k-diff pairs in the array. Here a k-diff pair is defined as an integer pair (i, j), where i and j are both numbers in the array and their absolute difference is k.

Answer:

class Solution {
public:
    int findPairs(vector<int>& nums, int k) {
        if(nums.size() < 2)
            return 0;
         //之前用map比较耗时
        /*map<int, int> mp;
        for(auto i : nums){
            if(mp.find(i) == mp.end()){
                mp.insert(make_pair(i,1));
            }
            else{
                ++mp.find(i)->second;
            }
        }*/
        int count = 0;
        vector<int> temp = nums;
        sort(temp.begin(), temp.end());
        if(k < 0)
            return 0;
        if(k==0) {
            for(int i=0;i<temp.size()-1;i++)
                //若数组下标为0且与下标为1的相等或者下标为i与i+1相等但i和i-1不等（避免重复计数eg:1,1,1）
                if(temp[i]==temp[i+1]&&(i==0||temp[i]!=temp[i-1])){
                    ++count;
                }
            return count;
        }
        /*for(auto beg = mp.begin(); beg != mp.end(); ++beg){
            temp.push_back(beg->first);
        }*/
        for(int i = 0; i < temp.size()-1; ++i){
            if(temp[i] != temp[i+1]){
                //binary_search:binary_search(data, data + N, key);查找范围:从data数组的第一个元素到第N-1个元素;如果找到key,返回true;没有找到,返回false.常用于查找元素是否存在.
                if(binary_search(temp.begin(), temp.end(), temp[i]+k))
                    ++count;
            }
        }
        return count;
        
}
};