在 ruby 中编写嵌套 "for" 的另一种有效方法？

Question

我有一个来自 HackerRank 的练习，可以很容易地解决如下问题：

def divisibleSumPairs(n, k, ar)
  validPairs = 0
  for i in 0..ar.size-1
    for j in i+1..ar.size-1
      validPairs += 1 if (ar[i]+ar[j]) % k == 0
    end
  end
  validPairs
end

n, k = gets.strip.split(' ')
n = n.to_i
k = k.to_i
ar = gets.strip
ar = ar.split(' ').map(&:to_i)
result = divisibleSumPairs(n, k, ar)
puts result;

但是嵌套的问题困扰着我。在 Ruby 中还有其他方法吗？

Answer 1

是的。您可以使用 Array class:

中的 combination 方法

a = [1,2,3,4,5]
a.combination(2).to_a #=> [[1, 2], [1, 3], [1, 4], [1, 5], [2, 3], [2, 4], [2, 5], [3, 4], [3, 5], [4, 5]]

然后你可以迭代这些对，所以代码应该看起来像那样（虽然运行还没有）：

def divisibleSumPairs(n, k, ar)
  validPairs = 0
  ar.combination(2).each do |pair|
    validPairs += 1 if (pair[0]+pair[1]) % k == 0
  end
  validPairs
end

在 ruby 中编写嵌套 "for" 的另一种有效方法？

Another efficient way to write a nested "for" in ruby?

ruby

algorithm

nested-loops