PTA-c语言 水仙花数

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

输入格式:

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

输出格式:

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

输入样例:

3

输出样例:

153
370
371
407

 代码块:

#include <stdio.h>
#include <math.h>
int pows(int a,int n){
    int k,i;
    k = a;
    for(i = 1;i < n;i++){
        a*=k;
    }
    return a;
}
int main(){
    
    int n = 0;
    int i,k,sum;
    int a;
    scanf("%d",&n);
    
    for(i = pow(10,n-1);i < pow(10,n);i++){
        sum = 0;
        k = i;
        while(k){
            a = k%10;
            sum += pows(a,n);
            k = k/10;
         }
                if(i == sum){
                    printf("%d\n",i);
                }
        
    }
}