532. 数组中的K-diff数对

https://leetcode-cn.com/problems/k-diff-pairs-in-an-array/

image.png

这一题可以用字典的方法

class Solution:
    def findPairs(self, nums: List[int], k: int) -> int:
        if k<0:
            return 0
        nums.sort()
        numsDict = {}
        for i in nums:
            if i in numsDict:
                numsDict[i] += 1
            else:
                numsDict[i] = 1

        ans = 0
        if k==0:
            for i in numsDict:
                if numsDict[i]>1:
                    ans += 1
        else:
            for i in numsDict:
                if i+k in numsDict:
                    ans += 1

        return ans

你可能感兴趣的:(532. 数组中的K-diff数对)