[JAVA]求21位水仙花数

题目

一个N位的十进制正整数，如果它的每个位上的数字的N次方的和等于这个数本身，则称其为花朵数。
例如：
当N=3时，153就满足条件，因为 1^3 + 5^3 + 3^3 = 153，这样的数字也被称为水仙花数（其中，“^{”表示乘方，5}3表示5的3次方，也就是立方）。
当N=4时，1634满足条件，因为 1^4 + 6^4 + 3^4 + 4^4 = 1634。
当N=5时，92727满足条件。
实际上，对N的每个取值，可能有多个数字满足条件。
程序的任务是：求N=21时，所有满足条件的花朵数。注意：这个整数有21位，它的各个位数字的21次方之和正好等于这个数本身。

如果满足条件的数字不只有一个，请从小到大输出所有符合条件的数字，每个数字占一行。因为这个数字很大，请注意解法时间上的可行性。要求在1分钟之内运算完毕。

/**
 * 求21位水仙花数
 */
public class Question4{
	//总位数，即题目中的N
	private static int end;
	
	//备忘录，npow[i][j]存的是：i的end次方乘以j得到的数
	private static BigInteger[][] npow;
	
	//用来排序
	private static TreeSet<String> ts = new TreeSet<>();

	
	public static void main(String[] args){
		long now = System.currentTimeMillis();
        beautiful_num(21);
        for(String item: ts) System.out.println(item);
        long last = System.currentTimeMillis();
        System.out.println("用时： "+(last-now)/1000.0+"s");
        
	}
	
	public static void beautiful_num(int num) {
		end = num;
		int[] pre = new int[10];
		BigInteger sum = BigInteger.ZERO;
		npow = new BigInteger[10][end+1];
		//计算备忘录
		for(int i=0;i<10;i++) {
			for(int j=0;j<end+1;j++) {
				npow[i][j] = BigInteger.valueOf(i).pow(end).multiply(BigInteger.valueOf(j));
			}	
		}
		recursion(pre, 0, end);
	}
	
	/**
	 * 递归体，N从0到9分配，每个数字可以被分配0-21次，所有数字加起来一共有21次分配机会
	 * @param pre 当前状态数组：pre[i]表示分配给数字i的次数
	 * @param N 当前正在分配的数字
	 * @param len 当前剩余位数
	 */
	public static void recursion(int[] pre, int N, int len) {
		//N==9说明分配已完成，开始检验是否符合条件
		if(N==9) {
			pre[N] = len;
			//首先检验当前pre组成的数是否为21位，滤掉不是21位的情况
			if(len>9 || len>end || len<0) return;
			BigInteger sum = BigInteger.ZERO;
			//是21位，则判断该21位数是否可以由pre[]组成
			for(int i=0;i<10;i++) {
//				System.out.println("i:"+i+"  pre[i]:"+pre[i]);
				sum = sum.add(npow[i][pre[i]]);
			}
			String temp = sum.toString(); //sum.toString()很耗时，一定要写出来
			if(temp.length()!=end) return;
			for(int i=0;i<10;i++) {
				int[] count = new int[10];
				for(int j=0;j<temp.length();j++) {
					count[temp.charAt(j)-'0']++;
				}
				for(int j=0;j<10;j++) {
					if(count[j]!= pre[j]) return;
				}
			}
			//找到符合条件的数，加到结果集ts里
			ts.add(temp);
			return;
		}
		for(int i=0;i<=len;i++) {
			pre[N] = i;
			recursion(pre, N+1, len-i);
		}
	}

程序在我i7-8750H的CPU上耗时6.6s左右，远小于题目要求时间。