Shivam Bajaj's submission of Dilemma | CodeChef