CodeChef solution