如何在 Ruby 中交错不同长度的数组

如果我想在 Ruby 中交错一组数组，并且每个数组的长度相同，我们可以这样做：

a.zip(b).zip(c).flatten

但是，如果数组的大小可以不同，我们如何解决这个问题呢？

我们可以做这样的事情：

def interleave(*args)
  raise 'No arrays to interleave' if args.empty?
  max_length = args.inject(0) { |length, elem| length = [length, elem.length].max }
  output = Array.new
  for i in 0...max_length
    args.each { |elem|
      output << elem[i] if i < elem.length
    }
  end
  return output
end

但是有没有更好的“Ruby”方式，也许使用 zip 或转置或类似的东西？

这是一个更简单的方法。它利用了您传递数组的顺序zip:

def interleave(a, b)
  if a.length >= b.length
    a.zip(b)
  else
    b.zip(a).map(&:reverse)
  end.flatten.compact
end

interleave([21, 22], [31, 32, 33])
# => [21, 31, 22, 32, 33]

interleave([31, 32, 33], [21, 22])
# => [31, 21, 32, 22, 33]

interleave([], [21, 22])
# => [21, 22]

interleave([], [])
# => []

请注意：这会删除所有nil's:

interleave([11], [41, 42, 43, 44, nil])
# => [11, 41, 42, 43, 44]