Apr 08

The sum of the squares of the first ten natural numbers is,
1^(2) + 2^(2) + ... + 10^(2) = 385

The square of the sum of the first ten natural numbers is,
(1 + 2 + ... + 10)^(2) = 55^(2) = 3025

Hence the difference between the sum of the squares of the first ten natural numbers and the square of the sum is 3025 − 385 = 2640.

Find the difference between the sum of the squares of the first one hundred natural numbers and the square of the sum.

My solution in Ruby:

def sum_and_square(j, k)
  tmp = 0
  for i in 1..100
    tmp += i**j
  end
  tmp**k
end
 
puts sum_and_square(1,2) - sum_and_square(2,1)