Exponentiation

import sys

sys.setrecursionlimit(10**6)
mod = 10**9+7
def fast_exp(a, b):
    if b == 0:
        return 1
    if b % 2 == 0:
        return fast_exp(a ** 2, b // 2) % mod
    return (a * fast_exp(a ** 2, b // 2)) % mod


n = int(input())
for i in range(n):
    a, b = map(int, input().split())
    print(fast_exp(a, b))

Test 1

Verdict: ACCEPTED

Test 2

Verdict: TIME LIMIT EXCEEDED

Test 3

Verdict: TIME LIMIT EXCEEDED