【PTA】习题4-6 水仙花数 (20分)

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

输入格式:

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

输出格式:

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

输入样例:

3

输出样例:

153
370
371
407

我的代码

#include<stdio.h>
#include<math.h>
int main(){
    int n;//几位水仙花数
    int i,num;
    double sum=0;
    scanf("%d",&n);
    int max=pow(10,n);
    for(i=pow(10,n-1);i<max;i++){
        num=i;
        sum=0; //做循环时，要注意每次都初始化条件！！
        while(num>0){    	
            sum+=pow(num%10,n);
            num/=10;
        }
        if(sum==i){printf("%d\n",i);
		}
    }
    return 0;
}

运行结果

测试点：最大N,输出4个 运行超时
其他测试点答案正确。

但是输入最大N ：7时，发现输出是对的。

修改

自己定义函数代替pow函数计算数字的n次幂。
pow(num%10,n);

#include<stdio.h>
#include<math.h>
int p(int a,int b);
int main(){
    int n;//几位水仙花数
    int i,num;
    double sum=0;
    scanf("%d",&n);
    int max=pow(10,n);
    for(i=pow(10,n-1);i<max;i++){
        num=i;
        sum=0;
        while(num>0){    	
            sum+=p(num%10,n);
            num/=10;
        }
        if(sum==i){printf("%d\n",i);
		}
    }
    return 0;
}
int p(int a,int b)
{
 int t=a;
 for(int i=1;i<b;i++)
  a=a*t;
  
 return a;
}

修改参考：
https://blog.csdn.net/bbyz1023/article/details/80392739