K-diff Pairs in an Array leetcode 532

最新推荐文章于 2022-02-09 17:35:51 发布

mmscan

最新推荐文章于 2022-02-09 17:35:51 发布

阅读量637

点赞数

CC 4.0 BY-SA版权

分类专栏：算法与面试文章标签： ruby leetcode 算法

本文链接：https://blog.youkuaiyun.com/tbestcc/article/details/60736048

算法与面试专栏收录该内容

16 篇文章

订阅专栏

本文针对一个特定的数组问题，提出了一种高效的解决方案。通过直接统计数组中各元素的出现次数来找出所有差值绝对值为k的不重复pair，避免了使用组合函数导致的时间复杂度过高。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目大意：给定一个数组，从中取出差值绝对值为k的pair，pair不能重复。

题目分析：如果调用combination之类的函数，会造成实际上的算法复杂度为O(n^2)，最后导致了TLE。因此改为直接统计数字在数组中出现的次数，然后根据 k 的值来进行不同的判断统计。

AC code（Ruby）：

def find_pairs(nums, k)
    if k < 0 || nums.length < 2
        return 0
    end
    count = 0
    h = Hash.new
    nums.each {|n| h[n] = (h.include? n) ? h[n] + 1 : 1 }
    if k == 0
        h.each_value {|v| count += 1 if v > 1 }
    elsif k > 0
        h.each_key {|key| count += 1 if h.include? key + k }
    end
    count
end