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>
int pow(int a,int b){
    int i,t=a;
    for(i=1;i<b;i++){
        a=a*t;
    }
     return a;
}
int main (){
    int i,n,N;
    int sum,num;
    scanf("%d",&N);
    n=pow(10,N);
    for(i=pow(10,N-1);i<n;i++){
        num=i;
        sum=0;
        while(num>0){
            sum=sum+pow(num%10,N);
            num=num/10;
        }
        if(sum==i){
        printf("%d\n",i);
        }
    }   
    return 0;
}

 注：这个pow函数必须自己写，如果用math头文件里的，PTA系统会判定超时。