ruby array flatten example

gistfile1.txt

def flatten(array)
  result = []
  array.each do |element|
    if element.class == Array
      flatten(element).each do |sub|
        result << sub
      end
    else
      result << element
    end
  end
  result
end

puts flatten([[1,2,[3]],4]) #=> [1,2,3,4]