https://realcoder.techss24.com/solved-write-a-function-hackerrank-problem/2634/