Array Mathematics Hackerrank Solution Python W3schools