Arrays Ds Hackerrank Solution Python