水仙花数代码如何减短运行时间

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

输入格式:

输入在一行中给出一个正整数N(3<=N<=7)。

输出格式:

按递增顺序输出所有N位水仙花数,每个数字占一行。
输入样例:3

输出样例:153
370
371
407

我的代码运行起来超时;我知道可能是n=7时循环次数太多造成的.有什么办法解决呢

include

include

int main()
{
int n,i,x;
scanf("%d", &n);
x=pow(10,n-1);
int y=pow(10,n);
int sum,a,z,flag;
for(i=x;i {
z=x;
sum=0;
flag=i;
while(z>0)
{
a=flag/z;
sum+=pow(a,n);
flag=flag%z;
z/=10;
}
if(sum==i) printf("%d\n", i);
}
return 0;
}

pow的效率很低,因为它是浮点运算,可以直接内联整数循环,甚至可以数组查表来代替。这样可以快很多。

int pow10[] = { 1, 10, 100, 1000, 10000, ... };

int y=pow(10,n);
->
int y=pow10[n];

感谢楼上的提醒,我自己写了个power函数就通过了

#include <stdio.h>

int mypow(int a, int ae) //不要用math的pow函数,因为是浮点运算,太慢了
{
    int powret;
    if (a == 0 || a == 1) //对于0和1的情况,没有必要计算
        powret = a;
    else {
        powret = 1;
        while (ae > 0)
        {
            powret = powret * a;
            ae--;
        }
    }

    return powret;
}

int split_and_sum(int i, int e) {
    int ret = 0;
    while (i > 0) {
        ret += mypow(i % 10, e);
        i /= 10;
    }

    return ret;
}

int main(void)
{
    int N;
    scanf("%d", &N);
    int min = 10, max;

    min = mypow(min, N - 1);//注意,3位数只有2个0
    max = min * 10;

    for (int i = min; i < max; ++i)
    {
        int sum = split_and_sum(i, N);
        if (sum == i)
            printf("%d\n", i);
    }

    return 0;
}