数组中的 k-diff 数对

简介: 我们拿到本题,读取题意要求在一组整数数组中,求出差值为k的数对对数k-diff。在思考如何解答该题之前,需要明确如下几点细节:nums数组元素都是整数索引位置i与位置j,不能相等k-diff数对关系:nums[i] - nums[j] = k -> nums[i] = nums[j] + k -〉 nums[i] - k = nums[j]k-diff数对,存在相同数对情况,但结果只取1次

一、题目描述:

题目内容

题目示例

题目解析

1 <= nums.length <= 104
-107 <= nums[i] <= 107
0 <= k <= 107

二、思路分析:
我们拿到本题,读取题意要求在一组整数数组中,求出差值为k的数对对数k-diff。在思考如何解答该题之前,需要明确如下几点细节:

nums数组元素都是整数
索引位置i与位置j,不能相等
k-diff数对关系:nums[i] - nums[j] = k -> nums[i] = nums[j] + k -〉 nums[i] - k = nums[j]
k-diff数对,存在相同数对情况,但结果只取1次

因此,我们的对题目中进行详细了解了,因为会排除重复的数对,我们很容易想哈希表来构建

方法一:构建哈希表

数对中重复场景如示例一中差值为k=1,(1,3) & (3,1)视为一种情况,则要定义两个哈希表来储存
哈希表可以通过字典k-value或者集合set(),本题无需考虑索引关系定义ans,numset两个集合
当 nums[i] > nums[j],则nums[j] = nums[i] - k在numset中,取最小的那一个则ans.add(nums[i]-k),
当 nuns[i] < nums[j],则nums[j] = nums[i] + k 在numset中,取较小的那一个则ans.add(nums[i])

根据上述思路,我们使用python代码能快速实现,代码如下:
class Solution(object):

def findPairs(self, nums, k):
    """
    :type nums: List[int]
    :type k: int
    :rtype: int
    """
    ans = set()
    numset = set()
    for num in nums:
        if num - k in numset:
            ans.add(num-k)
        if num + k in numset:
            ans.add(num)          
        numset.add(num)       
    return len(ans)

复制代码

方法二:双指针

首先对nums数组中的元素按照从低到高的顺序排列
在递增的数组中,由于双指针 i!=j,因此i指针一定是小于j的
枚举查找的判断的条件 nums[j] < nums[i]+k,指针j则往后移动
当nums[j] = nums[i] + k 时,则对数次数+1

根据上述思路,使用python代码实现,代码如下:
class Solution(object):

def findPairs(self, nums, k):
    """
    :type nums: List[int]
    :type k: int
    :rtype: int
    """
    nums.sort()
    ans = 0
    j = 0
    for i in range(len(nums)):
        if i == 0 or nums[i] != nums[i-1]:
            while j < len(nums) and (nums[j] < nums[i] + k or j <= i):
                j +=1                
            if j < len(nums) and nums[j] == nums[i] + k:
                ans +=1    
    return ans

相关文章
|
7月前
|
C++ Python
leetcode-56:合并区间
leetcode-56:合并区间
61 0
|
算法
arr = [1,2,5,8,9,10,20,30,40] 有一个从小到大排序好的数组,现在输入一个数,要求按照原来的规律插入到数组中
arr = [1,2,5,8,9,10,20,30,40] 有一个从小到大排序好的数组,现在输入一个数,要求按照原来的规律插入到数组中
113 0
|
1月前
将奇数数组与偶数数组合并为一个数组
【10月更文挑战第29天】将奇数数组与偶数数组合并为一个数组。
25 4
|
2月前
|
C++
Leetcode第56题(合并区间)
这篇文章介绍了LeetCode第56题“合并区间”的解题方法,通过排序和贪心策略合并重叠区间,并提供了C++的代码实现。
26 0
Leetcode第56题(合并区间)
|
4月前
|
算法
LeetCode第56题合并区间
LeetCode第56题"合并区间"的解题方法,通过排序区间并判断重叠后进行合并,有效解决了区间合并问题。
LeetCode第56题合并区间
|
5月前
|
JavaScript
JS 【详解】双指针排序 -- 数组合并后递增排序
JS 【详解】双指针排序 -- 数组合并后递增排序
33 0
数组求最大值和最小值的方法,数组写最大值和最小值的三元写法,数组取最小值,如何在数组中添加最后一个值,arr.push()添加数组放到后面,arr.unshift(‘red‘)添加的放到前面
数组求最大值和最小值的方法,数组写最大值和最小值的三元写法,数组取最小值,如何在数组中添加最后一个值,arr.push()添加数组放到后面,arr.unshift(‘red‘)添加的放到前面
|
7月前
|
算法 测试技术 C#
【最大公约数 排序】2344. 使数组可以被整除的最少删除次数
【最大公约数 排序】2344. 使数组可以被整除的最少删除次数
|
7月前
|
人工智能
PTA-将数组中的数逆序存放
将数组中的数逆序存放
59 0