題目描述
給定一個整數數組和一個整數 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)