PTA|《C语言程序设计实验与习题指导（第3版）》实验4-2-5 水仙花数 (20分)

题目

水仙花数是指一个N位正整数（N≥3），它的每个位上的数字的N次幂之和等于它本身。例如：$153=1{}^3+5^3+3^3$。 本题要求编写程序,计算所有N位水仙花数。

输入格式:
输入在一行中给出一个正整数N（3≤N≤7）。

输出格式:
按递增顺序输出所有N位水仙花数，每个数字占一行。

输入样例:

3

输出样例:

153
370
371
407

AC代码

#include<stdio.h>
#include<math.h>
int main(){
    int N,num,max,p[10];	
    scanf("%d",&N);
    num=pow(10,N-1);
/*pow(10,N)和(0,9)的幂值为固定值，存入变量中，避免在条件判断及叠加中多次调用pow函数降低效率*/
    max=pow(10,N);
    for(int i=0;i<10;i++)p[i]=pow(i,N);
    while(num<max){
        int sum=0,tmp=num;         
        for(int i=0;i<N;i++){
            sum+=p[tmp%10];
            tmp/=10;
            if(sum>num)break;
        }
        if(sum==num)printf("%d\n",num);
        num++;
    }
    return 0;   
}