2016-07-08 29 views
-1

我想交错两个阵列而不完全交替它们。例如,给出:不等间隔地交错两个红宝石阵列

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

你是否特意想要一个长度为1:2的数组的答案?或者,你是否想要一个更一般的答案,即两个任意长度的数组? – sawa

+0

任意长度请! –

+0

这就是我最初的想法。这只会让我的答案适合。 – sawa

回答

4
a.each_slice(2).zip(b).flatten 
#=> [1, 2, "a", 3, 4, "b", 5, 6, "c"] 
+0

[不起作用](https://repl.it/CaAq/4) – potashin

+0

未定义任何数组中存在多余元素时的期望行为。所以我可以同样声称它完美地工作。无论如何,你不觉得downvote有点太苛刻吗? :) –

+0

问题的阐述可能是模糊的,但OP的期望的结果,你已经张贴不要在你的代码产生的结果相匹配的结果,这是肯定的 - 所以我想downvote是正当的,无意冒犯,顺便 – potashin

0

你可以尝试以下方法(对于这两个例子中,你已经提供):

b.zip(a.each_slice(2)).map(&:reverse).flatten 
# => [1, 2, "a", 3, 4, "b", 5, 6, "c"] 

Demonstration

0

这适用于阵列a和任何尺寸的b(包括b.size > a.size)并确保合并数组中来自a的元素的序列在长度上最多不等于一个。当b.size > a.sizeb的序列也是如此。

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"]