一道C语言题目，多谢大佬了

Question

给你一个数字N（小于10000），
将每一个数位上的数求平方之后，再求和，记为M。

要求输出N+M的值。

例如：

输入：15，你计算12+52=26，则你应该输出：41

输入：12，你计算12+22=5，则你应该输出：17

Answer 1

同学你好！下面是我写的程序运行结果和源代码， 完全按照题目给的原理写的代码。 

结果1

结果2

结果3

代码展示：

#include <stdio.h>

int main() {

    int N = 0, M = 0;

    int ans = 0;

    printf("输入：");

    scanf("%d", &N);

    if(N >= 10000) {

        printf("N必须小于10000\n");

    }

    else{

        if(0 <= N && N < 10) {

            M = N*N;

            ans = N + M;

        }

        if(10 <= N && N < 100) {

            M = (N%10)*(N%10) + (N/10)*(N/10);

            ans = N + M;

        }

        if(100 <= N && N < 1000) {

            M = (N%10)*(N%10) + (N%100)/10*(N%100)/10 + (N/100)*(N/100);

            ans = N + M;

        }

        if(1000 <= N && N < 10000) {

            M = (N%10)*(N%10) + (N%100)/10*(N%100)/10 + (N%1000)/100*(N%1000)/100 + (N/1000)*(N/1000);

            ans = N + M;

        }

        printf("ans = %d", ans);

    }

}

Answer 2

#!/bin/bash
while true;do
read -p "请输入一个1~9999的整数:" N
if [ -z $N ];then
echo "输入错误"
else
if [ $N -ge 1 -a $N -le 9999 ];then
break
else
echo "输入错误"
fi
fi
done
echo $N |awk '{print substr($1,1,1)*substr($1,1,1)+substr($1,2,1)*substr($1,2,1)+substr($1,3,1)*substr($1,3,1)+substr($1,4,1)*substr($1,4,1)+$1}'
~