532. K-diff Pairs in an Array

题目

大意是查找到，差值为k的数据对的数量在一个数组中。

我的代码（未通过）

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