使用

print((a**b)%p)

会TLE，你应该使用

print(pow(a,b,p))

这个python的自带函数可以直接解决题目的问题$a^b ~mod~p$。