[PTA]习题4-6 水仙花数

Spring-_-Bear 的 CSDN 博客导航

---

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

输入格式:

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

输出格式:

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

输入样例:

3

输出样例:

153
370
371
407

来源：

来源：PTA | 程序设计类实验辅助教学平台
链接：https://pintia.cn/problem-sets/12/exam/problems/292

提交：

题解：

#include<stdio.h>

/*
 * 求 x 的 n 次方，自定义幂运算函数解决超时问题
 */
int customizePow(int x, int n) {
    int result = 1;

    for (int i = 1; i <= n; i++) {
        result *= x;
    }

    return result;
}

int main(void) {
    int N;
    scanf("%d", &N);

    int max = customizePow(10, N) - 1;
    for (int i = customizePow(10, N - 1); i <= max; i++) {
        // 每个数的每一个数字的对应 N 次幂之和
        int sum = 0;
        int number = i;

        while (number != 0) {
            int lastNumber = number % 10;
            sum += customizePow(lastNumber, N);
            number /= 10;
        }

        // 输出水仙花数
        if (sum == i) {
            printf("%d\n", sum);
        }
    }

    return 0;
}