balls

#!/usr/bin/ruby

$balls = [[:gold, :gold], [:silver, :gold], [:silver, :silver]]

$first_draw = {}
$first_draw[:gold] = 0
$first_draw[:silver] = 0
$first_draw[:total] = 0

$second_draw = {}
$second_draw[:gold] = 0
$second_draw[:silver] = 0
$second_draw[:total] = 0

for i in (0...1000000)
  box_num = rand(3)
  box = $balls[box_num]
  ball_num = rand(2)
  ball = box[ball_num]

  $first_draw[ball] += 1
  $first_draw[:total] += 1

  next if ball == :silver

  ball2 = box[1 - ball_num]

  $second_draw[ball2] += 1
  $second_draw[:total] += 1
end

puts "First draw gold:    #{$first_draw[:gold]} / #{$first_draw[:total]} = #{$first_draw[:gold].to_f/$first_draw[:total].to_f}"
puts "First draw solver:  #{$first_draw[:silver]} / #{$first_draw[:total]} = #{$first_draw[:silver].to_f/$first_draw[:total].to_f}"
puts "When first draw is gold..."
puts "Second draw gold:   #{$second_draw[:gold]} / #{$second_draw[:total]} = #{$second_draw[:gold].to_f/$second_draw[:total].to_f}"
puts "Second draw solver: #{$second_draw[:silver]} / #{$second_draw[:total]} = #{$second_draw[:silver].to_f/$second_draw[:total].to_f}"