Array Mathematics Hackerrank Solution Python Ide