Написать программу, позволяющую по последней цифре числа находить последнюю цифру его квадрата. - C (СИ)

Формулировка задачи:

Написать программу, позволяющую по последней цифре числа находить последнюю цифру его квадрата. Например, квадраты чисел 47 и 127 будут оканчиваться на 9.

Решение задачи: «Написать программу, позволяющую по последней цифре числа находить последнюю цифру его квадрата.»

Листинг программы

#include <stdio.h>
 
int EndDigit(int N)
{
    return N<10?N:N%10;
}
 
int main(int argc, char *args[])
{
    int N = 0;
    printf("Type N: ");
    scanf("%d", &N);
 
    N = EndDigit(N);
    N = EndDigit(N*N);
 
    printf("Result: %d", N);
}

Объяснение кода листинга программы

Подключение необходимых библиотек для работы с файлами и вводом/выводом данных.
Определение функции EndDigit, которая принимает целое число и возвращает его последнюю цифру. Если число меньше или равно 10, то оно возвращается без изменений, иначе возвращается остаток от деления числа на 10.
В функции main инициализируется переменная N со значением 0.
Выводится сообщение Type N: и осуществляется ввод числа с помощью функции scanf. Введенное число сохраняется в переменной N.
Значение переменной N передается в функцию EndDigit, которая возвращает последнюю цифру числа. Результат сохраняется в переменной N.
Значение переменной N передается в функцию EndDigit, которая возвращает последнюю цифру квадрата числа. Результат сохраняется в переменной N.
Выводится сообщение Result: и значение переменной N.