How to solve function program in hacker rank