compress, decompress

compress_decompress.rb

def decompress(args)
  ans = []
  args.each do |a|
    if a.is_a? Array
      ans.push a.last*a.first
    else
      ans.push a
    end
  end
  ans.join("")
end

def compress(args)
  tmp = []
  ans = []
  tmp.push args.shift
  while args.any?
    tmp.push args.shift unless ans.empty?
    if args.first == tmp.last
      tmp.push args.shift
    else
      if tmp.size == 1
        ans.push tmp.last
      else
        ans.push [tmp.size, tmp.last]
      end
      tmp.clear
    end
  end
  ans
end