蓝桥杯 21位的水仙花数

import java.math.BigInteger;
import java.util.Scanner;

public class Main{
public static int size;
public static int array[]={0,1,2,3,4,5,6,7,8,9};
//记录0-9的size次方
public static BigInteger powArray[]=new BigInteger[10];
//记录0-9的使用次数
public static int usedTimes[]=new int[10];
//i^N*j
public static BigInteger isPowSizeMultiply[][];
public static BigInteger MAX;//size位的数字能表示的最大值
public static BigInteger MIN;//size位的数字能表示的最小值

public static void init(){
//用于初始化powArray[],MAX,MIN
for(int i=0;i<10;i++){
//初始化powArray[]
powArray[i]=(new BigInteger(""+i)).pow(size);
}
MIN=(new BigInteger("10")).pow(size-1);//初始化最小值
MAX=(new BigInteger("10")).pow(size).add(new BigInteger("-1"));//初始化最大值

isPowSizeMultiply=new BigInteger[10][size+1];
for(int i=0;i<10;i++){
isPowSizeMultiply[i][0]=BigInteger.valueOf(0);
for(int j=1;j<size+1;j++){
isPowSizeMultiply[i][j]=isPowSizeMultiply[i][j-1].add(powArray[i]);
}
}
}

public static void exhaustion(int arrayIndex,int used,BigInteger current){
if(current.compareTo(MAX)>1){
//超过最大值，递归结束
return;
}

if(used==size){
//size位全部分配完毕
if(current.compareTo(MIN)<0){
//已获得的值小于最小值
return;
}
else{
String s=current+"";
int avaliableValueUsed[]=new int[10];
for(int i=0;i<s.length();i++){
avaliableValueUsed[s.charAt(i)-'0']++;
}
for(int i=0;i<10;i++){
if(usedTimes[i]!=avaliableValueUsed[i]){
return;
}
}
System.out.println(current);
return;
}
}
if(arrayIndex==0){
usedTimes[0]=size-used;
exhaustion(-1,size,current);
usedTimes[0]=0;
return;
}
if(current.add(isPowSizeMultiply[arrayIndex][size-used]).compareTo(MIN)<0){
return;
}

if(arrayIndex>=0){
for(int i=0;i<=size-used;i++){
if(current.add(isPowSizeMultiply[arrayIndex][i]).compareTo(MAX)>0){
return;
}
usedTimes[arrayIndex]=i;
exhaustion(arrayIndex-1,used+i,current.add(isPowSizeMultiply[arrayIndex][i]));
usedTimes[arrayIndex]=0;
}
}else{
return;//1-9已经分配完毕，不可再延伸
}
}

public static void main(String[] args){
long startTime=System.currentTimeMillis();//程序开始的时间
size=21;
init();
exhaustion(9,0,BigInteger.valueOf(0));
long endTime=System.currentTimeMillis();//程序结束的时间
System.out.println((endTime-startTime)/1000f+"秒");
}

}