Shahjalal Shohag's submission of K-Prob | CodeChef