【力扣日记】532 数组中的K-diff数对

题目描述

给定一个整数数组和一个整数 k, 你需要在数组里找到不同的 k-diff 数对。这里将 k-diff 数对定义为一个整数对 (i, j), 其中 i 和 j 都是数组中的数字，且两数之差的绝对值是 k.

数对 (i, j) 和数对 (j, i) 被算作同一数对。
数组的长度不超过10,000。
所有输入的整数的范围在 [-1e7, 1e7]。

算法思路

NAIVE

第一版：

class Solution:
    def findPairs(self, nums: List[int], k: int) -> int:
        nums.sort()
        se=set()
        n=0
        for i,j in enumerate(nums):
            if j+k>nums[-1]:break
            # if i>0 and j==nums[i-1]:continue
            if j+k in nums[i+1:]:
                if (j,j+k)in se:continue
                n+=1
                se.add((j,j+k))
        return n

第二版：

        nums.sort()
        # se=set()
        n=0
        for i,j in enumerate(nums):
            if j+k>nums[-1]:break
            if i>0 and j==nums[i-1]:continue
            if j+k in nums[i+1:]:
                # if (j,j+k)in se:continue
                n+=1
                # se.add((j,j+k))
        return n

IMPROVE

字典和集合都是利用哈希表，查找的时间复杂度都是O(1)。

class Solution:
    def findPairs(self, nums, k: int) -> int:
        if k<0:return 0
        d,s,n=set(),set(),0
        for i in nums:
            if i+k in d and (i, i + k) not in s:
                n+=1
                s.add((i,i+k))
            if i-k in d and (i - k, i) not in s:
                n+=1
                s.add((i-k,i))
            d.add(i)
        return n

执行用时 :152 ms, 在所有 Python3 提交中击败了56.30%的用户
内存消耗 :16.2 MB, 在所有 Python3 提交中击败了5.45%的用户

        if k<0:return 0
        saw = set()
        diff = set()
        for i in nums:
            if i-k in saw:
                diff.add(i-k)
            if i+k in saw:
                diff.add(i)
            saw.add(i)
        return len(diff)