Ha python. Написать программу, позволяющую по последней цифре числа определить последнюю цифру его квадрата

x = int(input())

print(x * x % 10)

Достаточно возвести цифру в квадрат и посмотреть её последнюю цифру этого квадрата