x = int(input())
print(x * x % 10)
Объяснение:
Достаточно возвести цифру в квадрат и посмотреть её последнюю цифру этого квадрата
