Can't iterate over Time objects in Ruby - ruby
I am writing an appointment form that will let the user choose a date. It will then take the date and check against a Google Calendar what time slots are available for that date within a range of 30 minutes time intervals from 10:00am to 5:00pm.
Within my Calendar class, I have an available_times method:
def available_times(appointment_date)
appointment_date_events = calendar.events.select { |event| Date.parse(event.start_time) == appointment_date }
conflicts = appointment_date_events.map { |event| [Time.parse(event.start_time), Time.parse(event.end_time)] }
results = resolve_time_conflicts(conflicts)
end
This method takes a date and grabs the start_time and end_time for each event on that date. It then calls resolve_time_conflicts(conflicts):
def resolve_time_conflicts(conflicts)
start_time = Time.parse('10:00am')
available_times = []
14.times do |interval_multiple|
appointment_time = (start_time + interval_multiple * (30 * 60))
available_times << appointment_time unless conflicts.each{ |conflict| (conflict[0]..conflict[1]).include?(appointment_time)}
end
available_times
end
A 'Can't iterate over Time' error is being thrown when I attempt to iterate over the conflicts array. I tried to call to_enum on the conflicts array but am still getting the same error.
All of the other questions I saw on SO were referencing the step method, which doesn't seem applicable to this case.
Update:
Thanks #caryswoveland and #fivedigit. I combined both of your answers, which were very helpful for different aspects of my solution:
def available_times(appointment_date)
appointment_date_events = calendar.events.select { |event| Date.parse(event.start_time) == appointment_date }
conflicts = appointment_date_events.map { |event| DateTime.parse(event.start_time)..DateTime.parse(event.end_time) }
results = resolve_time_conflicts(conflicts)
end
def resolve_time_conflicts(conflicts)
date = conflicts.first.first
start_time = DateTime.new(date.year, date.month, date.day, 10, 00).change(offset: date.zone)
available_times = []
14.times do |interval_multiple|
appointment_time = (start_time + ((interval_multiple * 30).minutes))
available_times << appointment_time unless conflicts.any? { |conflict| conflict.cover?(appointment_time)}
end
available_times
end
Exception
#fivedigit has explained why the exception was raised.
Other problems
You need any? where you have each:
appointment_times = []
#=> []
appointment = 4
#=> 4
conflicts = [(1..3), (5..7)]
#=> [1..3, 5..7]
appointment_times << 5 unless conflicts.each { |r| r.cover?(appointment) }
#=> nil
appointment_times
#=> []
appointment_times << 5 unless conflicts.any? { |r| r.include?(appointment) }
#=> [5]
appointment_times
#=> [5]
I suggest you covert appointment_time to a Time object, make conflicts and array of elements [start_time, end_time] and then compare appointment_time to the endpoints:
...unless conflicts.any?{ |start_time, end_time|
start_time <= appointment_time && appointment_time <= end_time }
Aside: Range#include? only looks at endpoints (as Range#cover? does) when the endpoints are "numeric". Range#include? need only look at endpoints when they are Time objects, but I don't know if Ruby regards Time objects as "numeric". I guess one could look at the source code. Anybody know offhand?
Alternative approach
I would like to suggest a different way to implement your method. I will do so with an example.
Suppose appointments were in blocks of 15 minutes, with the first block being 10:00am-10:15am and the last 4:45pm-5:00pm. (blocks could be shorter, of course, as small as 1 second in duration.)
Let 10:00am-10:15am be block 0, 10:15am-10:30am be block 1, and so on, until block 27, 4:45pm-5:00pm.
Next, express conflicts as an array of block ranges, given by [start, end]. Suppose there were appointments at:
10:45am-11:30am (blocks 3, 4 and 5)
1:00pm- 1:30pm (blocks 12 and 13)
2:15pm- 3:30pm (blocks 17, 18 and 19)
Then:
conflicts = [[3,5], [12,13], [17,19]]
You must write a method reserved_blocks(appointment_date) that returns conflicts.
The remaining code is as follows:
BLOCKS = 28
MINUTES = ["00", "15", "30", "45"]
BLOCK_TO_TIME = (BLOCKS-1).times.map { |i|
"#{i<12 ? 10+i/4 : (i-8)/4}:#{MINUTES[i%4]}#{i<8 ? 'am' : 'pm'}" }
#=> ["10:00am", "10:15am", "10:30am", "10:45am",
# "11:00am", "11:15am", "11:30am", "11:45am",
# "12:00pm", "12:15pm", "12:30pm", "12:45pm",
# "1:00pm", "1:15pm", "1:30pm", "1:45pm",
# "2:00pm", "2:15pm", "2:30pm", "2:45pm",
# "3:00pm", "3:15pm", "3:30pm", "3:45pm",
# "4:00pm", "4:15pm", "4:30pm", "4:45pm"]
def available_times(appointment_date)
available = [*(0..BLOCKS-1)]-reserved_blocks(appointment_date)
.flat_map { |s,e| (s..e).to_a }
last = -2 # any value will do, can even remove statement
test = false
available.chunk { |b| (test=!test) if b > last+1; last = b; test }
.map { |_,a| [BLOCK_TO_TIME[a.first],
(a.last < BLOCKS-1) ? BLOCK_TO_TIME[a.last+1] : "5:00pm"] }
end
def reserved_blocks(date) # stub for demonstration.
[[3,5], [12,13], [17,19]]
end
Let's see what we get:
available_times("anything")
#=> [["10:00am", "10:45am"],
# ["11:30am", "1:00pm"],
# [ "1:45pm", "2:15pm"],
# [ "3:00pm", "5:00pm"]]
Explanation
Here is what's happening:
appointment_date = "anything" # dummy for demonstration
all_blocks = [*(0..BLOCKS-1)]
#=> [ 0, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13,
# 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27]
reserved_ranges = reserved_blocks(appointment_date)
#=> [[3, 5], [12, 13], [17, 19]]
reserved = reserved_ranges.flat_map { |s,e| (s..e).to_a }
#=> [3, 4, 5, 12, 13, 17, 18, 19]
available = ALL_BLOCKS - reserved
#=> [0, 1, 2, 6, 7, 8, 9, 10, 11, 14, 15, 16, 20, 21, 22, 23, 24, 25, 26, 27]
last = -2
test = false
enum1 = available.chunk { |b| (test=!test) if b > last+1; last = b; test }
#=> #<Enumerator: #<Enumerator::Generator:0x00000103063570>:each>
We can convert it to an array to see what values it would pass into the block if map did not follow:
enum1.to_a
#=> [[true, [0, 1, 2]],
# [false, [6, 7, 8, 9, 10, 11]],
# [true, [14, 15, 16]],
# [false, [20, 21, 22, 23, 24, 25, 26, 27]]]
Enumerable#chunk groups consecutive values of the enumerator. It does so by grouping on the value of test and flipping its value between true and false whenever a non-consecutive value is encountered.
enum2 = enum1.map
#=> #<Enumerator: #<Enumerator: (cont.)
#<Enumerator::Generator:0x00000103063570>:each>:map>
enum2.to_a
#=> [[true, [0, 1, 2]],
# [false, [6, 7, 8, 9, 10, 11]],
# [true, [14, 15, 16]],
# [false, [20, 21, 22, 23, 24, 25, 26, 27]]]
You might think of enum2 as a "compound" enumerator.
Lastly, we convert the second element of each value of enum2 that is passed into the block (the block variable a, which equals [0,1,2] for the first element passed) to a range expressed as a 12-hour time. The first element of each value of enum2 (true or false) is not used, so so I've replaced its block variable with an underscore. This provides the desired result:
enum2.each { |_,a|[BLOCK_TO_TIME[a.first], \
(a.last < BLOCKS-1) ? BLOCK_TO_TIME[a.last+1] : "5:00pm"] }
#=> [["10:00am", "10:45am"],
# ["11:30am", "1:00pm"],
# [ "1:45pm", "2:15pm"],
# [ "3:00pm", "5:00pm"]]
The issue comes from this bit:
(conflict[0]..conflict[1]).include?(appointment_time)
# TypeError: can't iterate from Time
You're creating a range of times and then checking if appointment_time falls within the range. This is what causes the error you're experiencing.
Instead of include?, you should use cover?:
(conflict[0]..conflict[1]).cover?(appointment_time)
This assumes that conflict[0] is the earliest time.
Convert your range from a range of times to a range of integers:
range = (conflict[0].to_i..conflict[1].to_i)
Then use the === operator as you used the include?:
conflict === appointment_time
EDIT: You can also obviously convert appointment_time to integer and still use include? since the range is now just an integer range.
Related
Ruby iterate over hours
Let's say I have some user input with start and end hours: start = 09:00 end = 01:00 How do I display all the hours between those 2? So from 09 to 23, 0, and then to 1. There are easy cases: start = 01:00 end = 04:00 That's just a matter of ((start_hour.to_i)..(end_hour.to_i)).select { |hour| }
This can be solved with a custom Enumerator implementation: def hours(from, to) Enumerator.new do |y| while (from != to) y << from from += 1 from %= 24 end y << from end end That gives you something you can use like this: hours(9, 1).each do |hour| puts hour end Or if you want an Array: hours(9,1).to_a #=> [9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 0, 1]
You could do a oneliner (0..23).to_a.rotate(start_h)[0...end_h - start_h] def hours_between(start_h, end_h) (0..23).to_a.rotate(start_h)[0...end_h - start_h] end hours_between(1, 4) # [1, 2, 3] hours_between(4, 4) # [] hours_between(23, 8) # [23, 0, 1, 2, 3, 4, 5, 6, 7] Don't forget to sanitize the input (That they are number between 0 and 23) :) If you want the finishing hour use .. instead of ... => [0..end_h - start_h] If you care about performance or want something evaluated lazily you can also do the following (reading the code is really clear): (0..23).lazy.map {|h| (h + start_h) % 24 }.take_while { |h| h != end_h }
With a simple condition: def hours(from, to) if from <= to (from..to).to_a else (from..23).to_a + (0..to).to_a end end hours(1, 9) #=> [1, 2, 3, 4, 5, 6, 7, 8, 9] hours(9, 1) #=> [9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 0, 1] You could also use the shorter, but more cryptic [*from..23, *0..to] notation.
https://stackoverflow.com/a/6784628/3012550 shows how to iterate over the number of hours in the distance between two times. I would use that, and at each iteration use start + i.hours def hours(number) number * 60 * 60 end ((end_time - start_time) / hours(1)).round.times do |i| print start_time + hours(i) end
Ruby reducing a number array into start end range array
I have an array of numbers as below: [11, 12, 13, 14, 19, 20, 21, 29, 30, 33] I would like to reduce this array to: [[11,14], [19,21], [29,30], [33,33]] Identify consequent numbers in an array and push only the start and end of its ranges. How to achieve this?
Exactly some problem is solved to give an example for slice_before method in ruby docs: a = [0, 2, 3, 4, 6, 7, 9] prev = a[0] p a.slice_before { |e| prev, prev2 = e, prev prev2 + 1 != e }.map { |es| es.length <= 2 ? es.join(",") : "#{es.first}-#{es.last}" }.join(",") In your case you need to tweak it a little: a = [11, 12, 13, 14, 19, 20, 21, 29, 30, 33] prev = a[0] p a.slice_before { |e| prev, prev2 = e, prev prev2 + 1 != e }.map { |es| [es.first, es.last] }
Here's another way, using an enumerator with Enumerator#next and Enumerator#peek. It works for any collection that implements succ (aka next). Code def group_consecs(a) enum = a.each pairs = [[enum.next]] loop do if pairs.last.last.succ == enum.peek pairs.last << enum.next else pairs << [enum.next] end end pairs.map { |g| (g.size > 1) ? g : g*2 } end Note that Enumerator#peek raises a StopInteration exception if the enumerator enum is already at the end when enum.peek is invoked. That exception is handled by Kernel#loop, which breaks the loop. Examples a = [11, 12, 13, 14, 19, 20, 21, 29, 30, 33] group_consecs(a) #=> [[11, 12, 13, 14], [19, 20, 21], [29, 30], [33, 33]] a = ['a','b','c','f','g','i','l','m'] group_consecs(a) #=> [["a", "b", "c"], ["f", "g"], ["i", "i"], ["l", "m"]] a = ['aa','ab','ac','af','ag','ai','al','am'] group_consecs(a) #=> [["aa", "ab", "ac"], ["af", "ag"], ["ai, ai"], ["al", "am"]] a = [:a,:b,:c,:f,:g,:i,:l,:m] group_consecs(a) #=> [[:a, :b, :c], [:f, :g], [:i, :i], [:l, :m]] Generate an array of seven date objects for an example, then group consecutive dates: require 'date' today = Date.today a = 10.times.map { today = today.succ }.values_at(0,1,2,5,6,8,9) #=> [#<Date: 2014-08-07 ((2456877j,0s,0n),+0s,2299161j)>, # #<Date: 2014-08-08 ((2456878j,0s,0n),+0s,2299161j)>, # #<Date: 2014-08-09 ((2456879j,0s,0n),+0s,2299161j)>, # #<Date: 2014-08-12 ((2456882j,0s,0n),+0s,2299161j)>, # #<Date: 2014-08-13 ((2456883j,0s,0n),+0s,2299161j)>, # #<Date: 2014-08-15 ((2456885j,0s,0n),+0s,2299161j)>, # #<Date: 2014-08-16 ((2456886j,0s,0n),+0s,2299161j)>] group_consecs(a) #=> [[#<Date: 2014-08-07 ((2456877j,0s,0n),+0s,2299161j)>, # #<Date: 2014-08-08 ((2456878j,0s,0n),+0s,2299161j)>, # #<Date: 2014-08-09 ((2456879j,0s,0n),+0s,2299161j)> # ], # [#<Date: 2014-08-12 ((2456882j,0s,0n),+0s,2299161j)>, # #<Date: 2014-08-13 ((2456883j,0s,0n),+0s,2299161j)> # ], # [#<Date: 2014-08-15 ((2456885j,0s,0n),+0s,2299161j)>, # #<Date: 2014-08-16 ((2456886j,0s,0n),+0s,2299161j)> # ]]
This is some code I wrote for a project a while ago: class Array # [1,2,4,5,6,7,9,13].to_ranges # => [1..2, 4..7, 9..9, 13..13] # [1,2,4,5,6,7,9,13].to_ranges(true) # => [1..2, 4..7, 9, 13] def to_ranges(non_ranges_ok=false) self.sort.each_with_index.chunk { |x, i| x - i }.map { |diff, pairs| if (non_ranges_ok) pairs.first[0] == pairs.last[0] ? pairs.first[0] : pairs.first[0] .. pairs.last[0] else pairs.first[0] .. pairs.last[0] end } end end if ($0 == __FILE__) require 'awesome_print' ary = [1, 2, 4, 5, 6, 7, 9, 13, 12] ary.to_ranges(false) # => [1..2, 4..7, 9..9, 12..13] ary.to_ranges(true) # => [1..2, 4..7, 9, 12..13] ary = [1, 2, 4, 8, 5, 6, 7, 3, 9, 11, 12, 10] ary.to_ranges(false) # => [1..12] ary.to_ranges(true) # => [1..12] end It's easy to change that to only return the start/end pairs: class Array def to_range_pairs(non_ranges_ok=false) self.sort.each_with_index.chunk { |x, i| x - i }.map { |diff, pairs| if (non_ranges_ok) pairs.first[0] == pairs.last[0] ? [pairs.first[0]] : [pairs.first[0], pairs.last[0]] else [pairs.first[0], pairs.last[0]] end } end end if ($0 == __FILE__) require 'awesome_print' ary = [1, 2, 4, 5, 6, 7, 9, 13, 12] ary.to_range_pairs(false) # => [[1, 2], [4, 7], [9, 9], [12, 13]] ary.to_range_pairs(true) # => [[1, 2], [4, 7], [9], [12, 13]] ary = [1, 2, 4, 8, 5, 6, 7, 3, 9, 11, 12, 10] ary.to_range_pairs(false) # => [[1, 12]] ary.to_range_pairs(true) # => [[1, 12]] end
Here's an elegant solution: arr = [11, 12, 13, 14, 19, 20, 21, 29, 30, 33] output = [] # Sort array arr.sort! # Loop through each element in the list arr.each do |element| # Set defaults - for if there are no consecutive numbers in the list start = element endd = element # Loop through consecutive numbers and check if they are inside the list i = 1 while arr.include?(element+i) do # Set element as endd endd = element+i # Remove element from list arr.delete(element+i) # Increment i i += 1 end # Push [start, endd] pair to output output.push([start, endd]) end
[Edit: Ha! I misunderstood the question. In your example, for the array a = [11, 12, 13, 14, 19, 20, 21, 29, 30, 33] you showed the desired array of pairs to be: [[11,14], [19,21], [29,30], [33,33]] which correspond to the following offsets in a: [[0,3], [4,6], [7,8], [9,9]] These pairs respective span the first 4 elements, the next 3 elements, then next 2 elements and the next element (by coincidence, evidently). I thought you wanted such pairs, each with a span one less than the previous, and the span of the first being as large as possible. If you have a quick look at my examples below, my assumption may be clearer. Looking back I don't know why I didn't understand the question correctly (I should have looked at the answers), but there you have it. Despite my mistake, I'll leave this up as I found it an interesting problem, and had the opportunity to use the quadratic formula in the solution. tidE] This is how I would do it. Code def pull_pairs(a) n = ((-1 + Math.sqrt(1.0 + 8*a.size))/2).to_i cum = 0 n.downto(1).map do |i| first = cum cum += i [a[first], a[cum-1]] end end Examples a = %w{a b c d e f g h i j k l} #=> ["a", "b", "c", "d", "e", "f", "g", "h", "i", "j", "k", "l"] pull_pairs(a) #=> [["a", "d"], ["e", "g"], ["h", "i"], ["j", "j"]] a = [*(1..25)] #=> [ 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, # 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25] pull_pairs(a) #=> [[1, 6], [7, 11], [12, 15], [16, 18], [19, 20], [21, 21]] a = [*(1..990)] #=> [1, 2,..., 990] pull_pairs(a) #=> [[1, 44], [45, 87],..., [988, 989], [990, 990]] Explanation First, we'll compute the the number of pairs of values in the array we will produce. We are given an array (expressed algebraically): a = [a0,a1,...a(m-1)] where m = a.size. Given n > 0, the array to be produced is: [[a0,a(n-1)], [a(n),a(2n-2)],...,[a(t),a(t)]] These elements span the first n+(n-1)+...+1 elements of a. As this is an arithmetic progession, the sum equals n(n+1)/2. Ergo, t = n(n+1)/2 - 1 Now t <= m-1, so we maximize the number of pairs in the output array by choosing the largest n such that n(n+1)/2 <= m which is the float solution for n in the quadratic: n^2+n-2m = 0 rounded down to an integer, which is int((-1+sqrt(1^1+4(1)(2m))/2) or int((-1+sqrt(1+8m))/2) Suppose a = %w{a b c d e f g h i j k l} Then m (=a.size) = 12, so: n = int((-1+sqrt(97))/2) = 4 and the desired array would be: [['a','d'],['e','g'],['h','i'],['j','j']] Once n has been computed, constructing the array of pairs is straightforward.
How to refer the index of an element in Array#delete_if
I want to build a custom method Array#drop_every(n) (I know it's monkey patching, I am doing this for a homework), which returns a new array omitting every nth element: [4, 8, 15, 16, 23, 42].drop_every(2) # [4, 15, 23] I want to implement it with Array#delete_if, but by referring to the index and not to the element itself, (similar to each_index) something like this: def drop_every(step) self.delete_if { |index| index % step == 0 } end How do I do this? I don't insist on using delete_if, I also looked at drop_while and reject, other suggestions are welcome.
You can use with_index method that returns enumerator, filter your collection and then get rid of the indexes. class Array def drop_every(step) self.each.with_index.select { |_, index| index % step == 0 }.map(&:first) end end [4, 8, 15, 16, 23, 42].drop_every(2) # => [4, 15, 23]
def drop_every(step) reject.with_index { |x,i| (i+1) % step == 0 } end [4, 8, 15, 16, 23, 42].reject.with_index{|x,i| (i+1) % 2 == 0} # => [4, 15, 23] [4, 8, 15, 16, 23, 42].reject.with_index{|x,i| (i+1) % 3 == 0} # => [4, 8, 16, 23]
You could use the values_at method to selectively filter out indices which you want. class Array def drop_every(step) self.values_at(*(0...self.size).find_all{ |x| (x+1) % step != 0 }) end end The answer was accepted while I was typing it. I will post it anyways.
def drop_every step delete_if.with_index(1){|_, i| i.%(step).zero?} end
class Array def drop_every(step) self.each_slice(step).flat_map{|slice| slice[0..-2]} end end p [4, 8, 15, 16, 23, 42].drop_every(2) #=> [4, 15, 23]
I'd extend the Enumerable mixin instead: module Enumerable def drop_every(step) return to_enum(:drop_every, step) unless block_given? each.with_index(1) do |o, i| yield o unless i % step == 0 end end end (1..10).drop_every(3) { |a| p a } # outputs below 1 2 4 5 7 8 10
Every Other 2 Items in Array
I need a ruby formula to create an array of integers. The array must be every other 2 numbers as follows. [2, 3, 6, 7, 10, 11, 14, 15, 18, 19...] I have read a lot about how I can do every other number or multiples, but I am not sure of the best way to achieve what I need.
Here's an approach that works on any array. def every_other_two arr arr.select.with_index do |_, idx| idx % 4 > 1 end end every_other_two((0...20).to_a) # => [2, 3, 6, 7, 10, 11, 14, 15, 18, 19] # it works on any array every_other_two %w{one two three four five six} # => ["three", "four"]
array = [] #Change 100000 to whatever is your upper limit 100000.times do |i| array << i if i%4 > 1 end
This code works for any start number to any end limit i = 3 j = 19 x =[] (i...j).each do |y| x << y if (y-i)%4<2 end puts x this should work
For fun, using lazy enumerables (requires Ruby 2.0 or gem enumerable-lazy): (2..Float::INFINITY).step(4).lazy.map(&:to_i).flat_map { |x| [x, x+1] }.first(8) #=> => [2, 3, 6, 7, 10, 11, 14, 15]
here's a solution that works with infinite streams: enum = Enumerator.new do |y| (2...1/0.0).each_slice(4) do |slice| slice[0 .. 1].each { |n| y.yield(n) } end end enum.first(10) #=> [2, 3, 6, 7, 10, 11, 14, 15, 18, 19] enum.each do |n| puts n end
Single Liner: (0..20).to_a.reduce([0,[]]){|(count,arr),ele| arr << ele if count%4 > 1; [count+1,arr] }.last Explanation: Starts the reduce look with 0,[] in count,arr vars Add current element to array if condition satisfied. Block returns increment and arr for the next iteration. I agree though that it is not so much of a single liner though and a bit complex looking.
Here's a slightly more general version of Sergio's fine answer module Enumerable def every_other(slice=1) mod = slice*2 res = select.with_index { |_, i| i % mod >= slice } block_given? ? res.map{|x| yield(x)} : res end end irb> (0...20).every_other => [1, 3, 5, 7, 9, 11, 13, 15, 17, 19] irb> (0...20).every_other(2) => [2, 3, 6, 7, 10, 11, 14, 15, 18, 19] irb> (0...20).every_other(3) => [3, 4, 5, 9, 10, 11, 15, 16, 17] irb> (0...20).every_other(5) {|v| v*10 } => [50, 60, 70, 80, 90, 150, 160, 170, 180, 190]
How to interleave arrays of different length in Ruby
If I want to interleave a set of arrays in Ruby, and each array was the same length, we could do so as: a.zip(b).zip(c).flatten However, how do we solve this problem if the arrays can be different sizes? We could do something like: 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 But is there a better 'Ruby' way, perhaps using zip or transpose or some such?
Here is a simpler approach. It takes advantage of the order that you pass the arrays to 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([], []) # => [] Be warned: this removes all nil's: interleave([11], [41, 42, 43, 44, nil]) # => [11, 41, 42, 43, 44]
If the source arrays don't have nil in them, you only need to extend the first array with nils, zip will automatically pad the others with nil. This also means you get to use compact to clean the extra entries out which is hopefully more efficient than explicit loops 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 Here is a slightly more complicated version that works if the arrays do contain 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
Your implementation looks good to me. You could achieve this using #zip by filling the arrays with some garbage value, zip them, then flatten and remove the garbage. But that's too convoluted IMO. What you have here is clean and self explanatory, it just needs to be rubyfied. Edit: Fixed the booboo. def interleave(*args) raise 'No arrays to interleave' if args.empty? max_length = args.map(&:size).max output = [] max_length.times do |i| args.each do |elem| output << elem[i] if i < elem.length end end output end a = [*1..5] # => [1, 2, 3, 4, 5] b = [*6..15] # => [6, 7, 8, 9, 10, 11, 12, 13, 14, 15] c = [*16..18] # => [16, 17, 18] interleave(a,b,c) # => [1, 6, 16, 2, 7, 17, 3, 8, 18, 4, 9, 5, 10, 11, 12, 13, 14, 15] Edit: For fun def interleave(*args) raise 'No arrays to interleave' if args.empty? max_length = args.map(&:size).max # assumes no values coming in will contain nil. using dup because fill mutates args.map{|e| e.dup.fill(nil, e.size...max_length)}.inject(:zip).flatten.compact end interleave(a,b,c) # => [1, 6, 16, 2, 7, 17, 3, 8, 18, 4, 9, 5, 10, 11, 12, 13, 14, 15]