Array Mathematics Hackerrank Solution Python Compiler