leetcode 532数组中的K-DIFF

给定一个整数数组和一个整数K，找出不同的K-DIFF数对，K-DIFF数对指整数对的差值为K

class Solution:
    def findPairs(self, nums: List[int], k: int) -> int:
        if k<0:
            return 0
        saw=set()
        diff=set()
        for i in nums:
            if i-k in saw:#3-2=1，将1添加到diff中
                diff.add(i-k)
            if i+k in saw:#3+2=5，将5添加到diff中
                diff.add(i)
            saw.add(i)
        return len(diff)

如[3,1,4,1,5],k=2   输出为2，数组中有两个K-DIFF数对(1,3),(5,3)