题解 | #k-dis数对#

由于数组的长度考虑采用哈希的方式,只遍历一次数组,利用 set 对结果进行去重

#
# 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
#
# 
# @param nums int整型一维数组 
# @param k int整型 
# @return int整型
#
from collections import defaultdict
class Solution:
    def k_dis(self , nums: List[int], k: int) -> int:
        # write code here
        res = set()
        count = defaultdict(int)
        for num in nums:
            if num - k in count:
                res.add(tuple(sorted([num - k, num])))
            if num + k in count:
                res.add(tuple(sorted([num + k, num])))
            count[num] += 1
        return len(res)