如果我想在
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或转置或其他一些？
解决方法
如果源数组中没有nil,则只需要使用nils扩展第一个数组,zip将自动使用nil填充其他数组.这也意味着你可以使用compact来清理额外的条目,这有望比显式循环更有效
def interleave(a,*args)
max_length = args.map(&:size).max
padding = [nil]*[max_length-a.size,0].max
(a+padding).zip(*args).flatten.compact
end
这是一个稍微复杂的版本,如果数组包含nil,则可以使用
def interleave(*args)
max_length = args.map(&:size).max
pad = Object.new()
args = args.map{|a| a.dup.fill(pad,(a.size...max_length))}
([pad]*max_length).zip(*args).flatten-[pad]
end