#prepare the input which is a string containing new lines
def parse(input)
  data = []
  input.each_line do |line|
    #Game 1: 3 blue, 4 red; 1 red, 2 green, 6 blue; 2 green
    game_num = line.split(":").first.split(" ").last.to_i
    draws = line.chomp.split(":").last.split(";").map {|e| e.split(",") }
    h = {}
    h[game_num] = draws
    data << h
  end
  data
end

# result should a single string or integer
def calculate(data)
  result = 0
  data.each do |d|
    draws = d.values.first
    mins = {
      red: 0,
      blue: 0,
      green: 0
    }
    draws.each do |set|
      set.each do |draw|
        num, color = draw.strip.split(" ")
        mins[color.to_sym] = num.to_i if num.to_i > mins[color.to_sym]
      end
    end
    result += mins[:red] * mins[:blue] * mins[:green]
  end
  result
end