codechef solution in one short