#prepare the input which is a string containing new lines
def parse(input)
  data = []
  input.each_line do |line|
    winners, draw = line.split(":").last.split("|")
    winners =winners.split(" ").map {|entry| entry.strip.to_i }
    draw = draw.split(" ").map {|entry| entry.strip.to_i }
    data << [winners,draw]
  end
  data
end

# result should a single string or integer
def calculate(data)
  cards = []
  data.length.times.each do |i|
    cards << 1
  end

  data.each_with_index do |d,i|
    winners, draw = d
    matches = (winners & draw).length
    matches.times do |j|
      cards[i+j+1] += cards[i]
    end
  end

  cards.reduce(&:+)
end