532. 数组中的 k-diff 数对

exiaohu 于 2022-06-16 发布

题目链接: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))