hackerrank 3n 1 solution in python