leetcode刷题之数组(三)

1.数组中的k-diff数对:给定一个整数数组和一个整数 k, 你需要在数组里找到不同的 k-diff 数对。这里将 k-diff 数对定义为一个整数对 (i, j), 其中 i 和 j 都是数组中的数字,且两数之差的绝对值是 k。注意:1.数对 (i, j) 和数对 (j, i) 被算作同一数对。2.数组的长度不超过10,000。3.所有输入的整数的范围在 [-1e7, 1e7]。题目链接:https://leetcode-cn.com/problems/k-diff-pairs-in-an-array/
解法I:(好垃圾的解法by myself)

class Solution:
    def findPairs(self, nums: List[int], k: int) -> int:
        if k < 0:
            return 0
        nums1 = list(set(nums))
        if k == 0:
            return len(set([i for i in nums if nums.count(i)>= 2]))
        nums1.sort()
        count = 0
        for i in nums1:
            if i-k in nums1:
                count += 1
        return count

解法II:参考链接:https://leetcode-cn.com/problems/k-diff-pairs-in-an-array/solution/ha-xi-onzui-jian-dan-jie-fa-by-mai-mai-mai-mai-zi/

 def findPairs(nums,k):
        if k<0:
            return 0
        saw, diff = set(), set()
        
       	for i in nums:
            if i-k in saw:
                diff.add(i-k)
            if i+k in saw:
                diff.add(i)
            saw.add(i)
        return len(diff)
发布了59 篇原创文章 · 获赞 8 · 访问量 1万+

猜你喜欢

转载自blog.csdn.net/Galen_xia/article/details/103720258