Smart Interview Basic Hackerrank Logic