【数组中的 k-diff 数对】python刷题记录

我明天再来学Web渗透

于 2024-07-25 22:38:56 发布

阅读量31

点赞数

文章标签： python 开发语言算法数据结构 leetcode

本文链接：https://blog.csdn.net/m0_73629042/article/details/140701450

版权

R2-哈希表。

有点easy的感觉

class Solution:
    def findPairs(self, nums: List[int], k: int) -> int:
        #查找对的方式是查找x+k，不查找x-k是避免查找重复
        #此外，需要注意k=0的问题
        mp={}
        for x in nums:
            if x in mp:
                mp[x]+=1
            else:
                mp[x]=1
        ret=0
        for x,cnt in mp.items():
            #k=0的情况处理
            if k==0:
                #一个数是否出现2次,出现就加1
                ret+=(cnt>=2)
            else:
                ret+=(x+k in mp)
        return ret