【问题标题】:Understanding two sums leetcode problem (Ruby)理解二和leetcode问题(Ruby)
【发布时间】:2020-05-01 22:57:02
【问题描述】:

我正在处理Leetcode Two Sums 问题:“给定一个整数数组,返回两个数字的索引,使它们相加为特定目标。”。这就是我所拥有的:

def two_sum(nums, target)
  hash = {}
  nums.each_with_index do |num, index|
    diff = target - num
    if hash[diff]
      return [hash[diff],index]
    else
      hash[num] = index
    end
  end
end

代码有效,但是,我不太确定为什么会这样。

所以我知道在每个语句中它都会遍历数字并找到差异。例如,

nums = [4,2,5,1]
target = 6

在第一个循环中,差值是 6-2 = 4。但是哈希显然是空的,所以它会将 num 注册为键,以当前索引作为值。因此哈希是,

hash = {
  4: 0
}

在第二个循环中,差异为 6-4 = 2。hash[4] 为 nil,因此它将当前的 num 和索引添加到字典中。

hash = {
  4: 0
  2: 1
}

像这样,这不会一直将 nums 添加到哈希中,因为至少在这种情况下没有匹配的键值对吗?

也许我把事情复杂化了。如果有人可以 eli5,我将不胜感激。谢谢!

【问题讨论】:

  • 如果没有一对数字总和为target,则返回nums,因此您可以添加nil 作为倒数第二行。您也可以将 if/else/end 结构替换为 return [hash[diff], index] if hash.key?(diff); hash[num] = index
  • 短而甜,虽然效率不高:nums.each_index.to_a.combination(2).find { |i,j| nums[i]+nums[j] == target }

标签: ruby


【解决方案1】:

诀窍是我们使用 number 作为键将值添加到哈希中:

hash[num] = index

但使用 diff 作为键提取它:

if hash[diff]

所以如果你有输入:

nums = [4,2,5,1]
target = 6

然后在第一步中,差值是 6 - 4 = 2,映射中没有键 2(差异),我们将键 4(数字)添加到映射中。 第二步,差是6 - 2 = 4,并且map中已经有一个key 4(diff),所以我们返回值。

【讨论】:

  • 天哪,这太有道理了!我觉得很笨哈哈。谢谢你们清理它!
猜你喜欢
  • 1970-01-01
  • 2020-02-16
  • 2020-04-24
  • 1970-01-01
  • 2020-06-13
  • 1970-01-01
  • 2019-05-13
  • 1970-01-01
  • 2019-10-29
相关资源
最近更新 更多