优化这个红宝石代码

Question

所以这段代码将计算其差值为K的数字对的总数。这是天真的方法，我需要优化它。建议？

test = $stdin.readlines 

input = test[0].split(" ") 

numbers = test[1].split(" ") 

N = input[0] 
K = input[1] 

count = 0 

for i in numbers 
    current = i.to_i 
    numbers.shift 
    for j in numbers 
     difference = (j.to_i - current).abs 
     if (difference == K) 
      count += 1 
     end 
    end 
end 

puts count 

Answer 1

本来对你给出一些输入和输出的例子不错，但我认为这是正确的。

require 'set' 

def count_diff(numbers, difference) 
    set = Set.new numbers 
    set.inject 0 do |count, num| 
    set.include?(num+difference) ? count+1 : count 
    end 
end 


difference = gets.split[1].to_i 
numbers  = gets.split.map { |num| num.to_i } 

puts count_diff(numbers, difference) 

Answer 2

未经检验的，希望实际的Ruby代码

的文档集：http://www.ruby-doc.org/stdlib/libdoc/set/rdoc/classes/Set.html

require 'set' 
numbers_set = Set.new 
npairs = 0 

numbers.each do |number| 
    if numbers_set.include?(number + K) 
     npairs += 1 
    end 
    if numbers_set.include?(number - K) 
     npairs += 1 
    end 
    numbers_set.add(number) 
end 

Answer 3

我不知道红宝石所以我只是给你的大创意：

1. 获取列表
2. 保留布尔值如果在列表中的列表中存在
3. 回路数阵列（称之为arr），标志着关闭数为真，看看是否arr[num-K]和/或arr[num+K]都是如此，num是在你的列表中的号码

它使用了相当多的内存，但这样的另一种方法是做到以下几点：

1. 保持一个哈希表从一个整数n整数count
2. 通过你的L到达ist，将num+K和num-K添加到哈希映射中，相应地增加count
3. 查看您的列表并查看num是否在哈希映射中。如果是，通过count

Answer 4

有人增加您的柜台删除他的帖子，或者他的帖子被删除了...他有最好的解决办法，那就是：

test = $stdin.readlines 
input = test[0].split(" ") 
numbers = test[1].split(" ") 
K = input[1] 
count = 0 
numbers.combination(2){|couple| couple.inject(:-).abs == K ? count++} 
puts count 

，你甚至不需要N.