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

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

输入格式:

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

输出格式：

在一行中输出部分和的值，精确到小数点后两位。题目保证计算结果不超过双精度范围。

输入样例1:

3

输出样例1：

153
370
371
407

---

答案代码块

#include <stdio.h>
//本题不能用pow函数否则运算时间会超时 
int p(int z,int n);
int main(){
    int n,x=1,y=0,sum=0,z=0;
    int i;
    scanf("%d",&n);
    if(n>7||n<3) return 0;
	x=p(10,n-1);
    for(i=x;i<x*10;i++){
        y=i;
        sum=0;
        while(y>0){
            z=y%10;
            sum+=p(z,n);//给p求次方 
            y/=10;
        }
        if(sum==i){
			printf("%d\n",i);
		}
    }
    
    return 0;
}

int p(int z,int n){
	int i;
	int b;
	b=z;
	for(i=1;i<n;i++){
		z*=b;
	}
	return z;
}