呓语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

付费支持

由于本网站没有广告和任何形式的收入来源,希望获得您的资助。每篇技术性文章和每期shellcasts视频定价人民币1元,在您付费后可以任意观看和下载。

可以使用支付宝手机钱包扫描下方的二维码进行付款操作或者用支付宝转帐给richard.ma.19850509@gmail.com,谢谢!

Fork me on GitHub