532. K-diff Pairs in an Array

题目

大意是查找到,差值为k的数据对的数量在一个数组中。
532. K-diff Pairs in an Array

我的代码(未通过)

class Solution:
    def findPairs(self, nums: List[int], k: int) -> int:
        nums.sort()
        return len(set([(x,y) for x,y in itertools.combinations(nums,2) if abs(x-y)==k]))

对数据两两排列,判断绝对值=k,删除重复元素,计算长度,但当大量数据处理的时候,超时了。

一行代码(同时也是最高效的)

如果k>0:
注意:len(set(nums)&{n+k for n in nums})
这是两个集合进行与运算!
如果k=0:
计算有重复数量的元素的个数。

class Solution:
     def findPairs(self, nums, k):
        return len(set(nums)&{n+k for n in nums}) if k>0 else sum(v>1 for v in collections.Counter(nums).values()) if k==0 else 0