LeetCode: K-diff Pairs in an Array

Problem:

Given an array of integers and an integer k, you need to find the number of unique k-diff pairs in the array. Here a k-diff pair is defined as an integer pair (i, j), where i and j are both numbers in the array and their absolute difference is k.

题意：
给定一个数组nums和一个数k，判断数组中有多少个组合使得两个数的差为给定的数k，不考虑顺序和重复的数字

方法：

如果k不等于零的话，那就是nums和nums数组每个数加k集合的交，如果k等于零的话，那就统计数组中相同的数字即可

代码：

import collections
class Solution(object):
    def findPairs(self, nums, k):
        """
        :type nums: List[int]
        :type k: int
        :rtype: int
        """
        if k > 0:
            return len(set(nums) & set(n + k for n in nums))
        elif k == 0:
            return sum(v > 1 for v in collections.Counter(nums).values())
        else:
            return 0