Euler171 partial

k = gets.chomp.to_i
final = []
for i in 1..k
   testvar = 0
   n = i
   e = 0
    while n > 0
        (n % 10) < e ? n = 0 : (e = (n % 10))
        testvar +=  e**2
        n /= 10
    end
    final << i if ((Math.sqrt(testvar) % 1) == 0)
end

puts final.inject(0, :+)