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