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 
int p(int a,int n);//**用math.h会超时的**
int main(void)
{
    int n;
    scanf("%d",&n);
    int i;
    for(i=p(10,n-1);i<p(10,n);i++)
    {
        int sum=0,a=i;
        while(a!=0)
        {
            sum+=p(a%10,n);
            a/=10;
        }
        if(sum==i) printf("%d\n",i);
    }
    return 0;
}

int p(int a,int n)
{
    int i,sum=1;
    for(i=0;i<n;i++)
        sum*=a;
    return sum;
}