Вообще, python отлично работает с длинной арифметикой, поэтому можно спокойно написать (n - число, a - степень):
n, a = map(int, input().split())
print((n ** a) % 10)
Можно совершить две оптимизации:
1. При работе с модулем показатель степени можно взять по модулю. Таким образом, это ускорит работу при больших n:
n, a = map(int, input().split())
print(((n % 10) ** a) % 10)
2. Можно отдельно написать двоичное возведение в степень. Код с ним прикладывать не буду, подробнее тут: http://e-maxx.ru/algo/binary_pow
Примечание: ввод через enter осуществим просто как:
n = int(input())
a = int(input())