题目链接:532. 数组中的 k-diff 数对
先排序,然后判断,如果一个数 $n$,及 $n+k$ 都出现在数组中,那它就是一个符合条件的数对。
from collections import Counter
from typing import List
class Solution:
def findPairs(self, nums: List[int], k: int) -> int:
if k == 0:
return sum(1 if v > 1 else 0 for v in Counter(nums).values())
nums = set(nums)
return sum(1 if num + k in nums else 0 for num in sorted(nums))