adding
#include <cmath>
and changing
while(r <= __lg(k)) {
to
while(r <= log2(k)) {
fixed the problem of TLE. I assume the answer would also be correct but I have not submitted; it does pass the sample, though.

You might want to further research __lg().