Nie wiem, nie znam się, ten kod jest napisany w javie?
Używaj ideone, wtedy będzie wszystko wiadome, jaki to język i dla jakich danych testujesz kod.
No właśnie, dla jakich danych testujesz kod? Jeśli dla przykładu ze strony z opisem to nic dziwnego że masz TLE xD
Testów może być 20000, a liczby z przedziału od 1 do 10^6, to teraz tak przetestuj swój kod dla 20000 przypadków z a = 1 i b = 10^6.