solve me first hackerrank solution in python