simple factorization

factorize.rb

def factorize(num)
  ans = Hash.new(0)

  while true
    prime_factor = find_prime_factor(num)
    break if 0 == prime_factor

    ans[prime_factor] += 1
    num /= prime_factor
  end
  ans
end

def find_prime_factor(num)
  (2..num).each do |i|
    return i if 0 == num % i
  end
  0
end

p factorize(ARGV.first.to_i)