7-4 JAVA-水仙花数（Java for PTA）

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

输入格式:

输入一个正整数N（3≤N≤7）。

输出格式:

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

输入样例:

在这里给出一组输入。例如：

3

输出样例:

在这里给出相应的输出。例如：

153
370
371
407

import java.util.Scanner;

public class Main {
    public static void main(String[] args) {
        Scanner sc = new Scanner(System.in);
        int n = sc.nextInt();
        
        int[] powers = new int[10];
        for (int i = 0; i < 10; i++) {
            powers[i] = (int)Math.pow(i, n); // 预先计算各个数字的n次幂，防止超时
        }

        int start = (int)Math.pow(10, n - 1);
        int end = (int)Math.pow(10, n);

        for (int i = start; i < end; i++) {
            int temp = i;
            int sum = 0;
            while (temp > 0) {
                int digit = temp % 10;
                sum += powers[digit]; // 直接使用预先计算好的n次幂结果
                temp /= 10;
            }
            
            if (sum == i) {
                System.out.println(i);
            }
        }
    }
}

?

?