我想交错两个数组,而不是完全交替它们。例如,给定:
a = [1,2,3,4,5,6,7,8,9]
b = ['a','b','c']
我希望收到f = [1,2,'a',3,4,'b',5,6,'c',7,8,9]
作为输出。对于b
中的每两个元素,我需要a
中的一个元素。
解决方案似乎是:
a = [1,2,3,4,5,6,7,8,9]
b = ['a','b','c']
even, odd = a.partition.each_with_index{ |_, i| i.even? }
res = even.zip(odd,b)
res = res.flatten.reject(&:blank?)
#=> [1, 2, "a", 3, 4, "b", 5, 6, "c", 7, 8, 9]
但我对这个解决方案并不感到兴奋。有什么想法吗?
答案 0 :(得分:4)
a.each_slice(2).zip(b).flatten
#=> [1, 2, "a", 3, 4, "b", 5, 6, "c"]
答案 1 :(得分:0)
您可以尝试以下操作(对于您提供的两个示例):
b.zip(a.each_slice(2)).map(&:reverse).flatten
# => [1, 2, "a", 3, 4, "b", 5, 6, "c"]
答案 2 :(得分:0)
这适用于任何大小的数组a
和b
(包括b.size > a.size
),并确保合并数组中a
的元素序列的长度不同最多一个。 b
时来自b.size > a.size
的序列也是如此。
def insert_em(a,b)
b.size <= a.size ? recurse(a,b) : recurse(b,a)
end
def recurse(a,b)
i = a.size/(b.size+1)
b.size == 1 ? a[0,i] + b + a[i..-1] :
[*a[0,i], b.first].concat(recurse(a[i..-1], b[1..-1]))
end
a = [1,2,3,4,5,6,7,8,9]
insert_em a, ['a']
#=> [1, 2, 3, 4, "a", 5, 6, 7, 8, 9]
insert_em a, ['a','b']
#=> [1, 2, 3, "a", 4, 5, 6, "b", 7, 8, 9]
insert_em a, ['a','b','c']
#=> [1, 2, "a", 3, 4, "b", 5, 6, "c", 7, 8, 9]
insert_em a, ['a','b','c','d']
#=> [1, "a", 2, 3, "b", 4, 5, "c", 6, 7, "d", 8, 9]
insert_em a, ['a','b','c','d','e']
#=> [1, "a", 2, "b", 3, "c", 4, 5, "d", 6, 7, "e", 8, 9]
insert_em a, ['a','b','c','d','e','f']
#=> [1, "a", 2, "b", 3, "c", 4, "d", 5, "e", 6, 7, "f", 8, 9]
insert_em a, ['a','b','c','d','e','f','g']
#=> [1, "a", 2, "b", 3, "c", 4, "d", 5, "e", 6, "f", 7, "g", 8, 9]
insert_em a, ['a','b','c','d','e','f','g','i']
#=> [1, "a", 2, "b", 3, "c", 4, "d", 5, "e", 6, "f", 7, "g", 8, "i", 9]
insert_em a, ['a','b','c','d','e','f','g','i','j']
#=> ["a", 1, "b", 2, "c", 3, "d", 4, "e", 5, "f", 6, "g", 7, "i", 8, "j", 9]
insert_em a, ['a','b','c','d','e','f','g','i','j','k']
#=> ["a", 1, "b", 2, "c", 3, "d", 4, "e", 5, "f", 6, "g", 7, "i", 8, "j", 9, "k"]
insert_em a, ['a','b','c','d','e','f','g','i','j','k','l']
#=> ["a", 1, "b", 2, "c", 3, "d", 4, "e", 5, "f", 6, "g", 7, "i", 8, "j", 9, "k", "l"]