【力扣日記】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)