呓语Beta 2.0 首页 镜头后 灶台前 捐赠者名单 有趣的小站

leetcode 532 K-diff Pairs In An Array

文档信息

leetcode 532 K-diff Pairs In An Array

解法一

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])) # item+k有多少元素在nums集合中,求交集的个数就是数对的数量
        if k == 0:
            return sum(v > 1 for v in collections.Counter(nums).values()) # 出现次数大于1的元素个数,重复过的元素就是数对的数量
        else:
            return 0

note

捐赠支持

可以使用支付宝转帐给richard.ma.19850509#gmail.com(请将#替换为@),谢谢!

Fork me on GitHub