I wrapped my solution in function  and added several test cases handling and submitted it to http://www.spoj.com/problems/KBASEEN/

Got WA

Maybe you try to do so

I am interested whether you will be AC