水仙花数计算

前段时间在CSDN论坛里面有个热帖，说的是水仙花数的计算方法（http://topic.csdn.net/u/20100819/21/6ea6612b-5d90-43fc-97a9-cdbccf3b2c13.html）。

 

 

题目是：

求解21位水仙花数的解法，要求在3分钟之内计算完成
水仙花数：N位整数，它等于各个位的数字的N次方之和，
例如有一个N位数字，a1a2a3a4.....aN = a1^N +a2^N+......aN^N

 

 

我的解决方法中 21位数字的计算时间为 16-17s，

 

现将源码及注释放出来：

 

 

import java.math.BigInteger; import java.util.Arrays; /** * 计算水仙花数 * <p> * 水仙花数：N位整数，它等于各个位的数字的N次方之和，例如有一个N位数字，a1a2a3a4.....aN = a1^N +a2^N+......aN^N * </p> * <br>算法原理： * <ol> * <li> * 定义：各个位的数字的N次方之和为 <b>水仙花和</b>。 * </li> * <li> * 对于任意两个N位数字，若数字0-9分别在它们中出现的次数相同，则这两个数字的水仙花和相同，即形如 * 123，321，213这样的数字的水仙花和相同，也就是说，它们之中至多只能有一个数字是水仙花数。 * </li> * <li> * 对于N位数字，最大的数字为10^N - 1， * 因此可以通过(10^N - 1) % (9 ^ N) 计算出可能为水仙花数的数字中9至多出现的次数。 * </li> * <li> * 对于N位数字，最小的数字为10^(N - 1)， * 注意到，对于任意水仙花和小于N位数字的最小值的数字，将该数的任意一位替换成更小的数字，也不能构成 * 水仙花数。 * </li> * <ol> * <p> * 综上所述，只需要对0-9进行N位的无序排列组合，并计算水仙花和，然后再与其各自出现次数比较，若0-9 * 出现次数相同，则该数字为水仙花数。 * </p> * */ public class Flower { /** 存储0-9的N次方 */ private static BigInteger[] powerNums = new BigInteger[10]; public static void main(String[] args) { int len = 5; if(args.length != 0 && args[0] != null){ len = Integer.parseInt(args[0]); } System.out.println("length:/t" + len); long time = System.nanoTime(); int total = find(len); time = System.nanoTime() - time; System.out.println("time:/t" + time / 1000000000.0 + "s"); System.out.println("total:/t" + total); } /** * 查找水仙花数 * @param n 位数 * @return 返回找到的水仙花数总量 */ public static int find(int n){ //若位数小于1则返回 if(n < 1){ return 0;} //水仙花数总量 int total = 0; //初始化 for(int i = 0; i <=9 ; i++){ powerNums[i] = BigInteger.valueOf(i).pow(n); } // 存储0-9的出现次数 int[] showTimes = new int[10]; // n位的最小数字 对于n=3,即100 //对于1位数字，最小的数字是 0 BigInteger minNum = n == 1 ? BigInteger.ZERO : BigInteger.TEN.pow(n - 1); // n位的最大数字 对于n=3,即999 BigInteger maxNum = BigInteger.TEN.pow(n).subtract(BigInteger.ONE); // 数字9出现的最大次数 int nineMaxShowTimes = maxNum.divide(powerNums[9]).intValue(); showTimes[9] = nineMaxShowTimes; showTimes[8] = n - nineMaxShowTimes; //指向当前最小数字，最小数字为0时会进行回溯，最终仍会指向当前最小数字 //n=1时会有特殊情况 int index = showTimes[8] == 0 ? 9 : 8; //回溯时0的出现次数 int moveNum = 0; //当前水仙花 BigInteger sum = sumShowTimes(showTimes); //是否进行判断：当前数字是否为水仙花数（回溯时不进行） boolean checkFlag = true; //是否进行判断：当前数字小于最小数字minNum boolean checkMinFlag = true; while(true){ //进行判断 if(checkFlag){ checkFlag = false; if(sum.compareTo(minNum) >= 0){ if(sum.compareTo(maxNum) <= 0 && Arrays.equals(showTimes,getCheckShowTimes(sum))){ //找到水仙花数 System.out.println(sum); total ++; } } else if(checkMinFlag){ checkMinFlag = false; if(checkMinValue(showTimes, minNum, n)){ //若showTimes形如 //[0, 1, 2, 3, 4, 5, 6, 7, 8, 9] //------------------------------ //[0, 0, k, n-k, 0, 0, 0, 0, 0, 0] //则进行比较，若水仙花和小于minNum，则后面的数字不可能是水仙花数 System.out.println("break:/t" + Arrays.toString(showTimes)); System.out.println("break:/t" + sum); break; } } } if(index > 0 && showTimes[index] > 0){ //showTimes进行变化：将当前一个最小数字替换为更小数字 //若当前最小数字为0，则将除0外一个最小数字替换为更小数字，并将0替换为这个更小数字 showTimes[index] --; sum = sum.subtract(powerNums[index]); index --; showTimes[index] += 1; sum = sum.add(powerNums[index]); if(moveNum != 0 && index != 0){ //回溯后 showTimes[index] += moveNum; sum = sum.add( powerNums[index].multiply( BigInteger.valueOf(moveNum) ) ); moveNum = 0; showTimes[0] = 0; //回溯后进行最小值判断 checkMinFlag = true; } //showTimes变化后进行当前数字是否为水仙花数判断 checkFlag = true; continue; }else if(showTimes[0] == n){ //若0的showTimes等于n，即所有数字都为0，则循环结束 System.out.println("break:/t" + Arrays.toString(showTimes)); System.out.println("break:/t" + sum); break; } //回溯，找到showTimes不为零的最小的数，并记录回溯前0的showTime //回溯期间不需要判断 index ++; moveNum = showTimes[0]; } return total; } /** * 根据各个数字的出现次数计算总值 * @param showTimes * @return */ private static BigInteger sumShowTimes(int[] showTimes){ BigInteger sum = BigInteger.ZERO; for(int i = 0; i < showTimes.length; i ++){ sum = sum.add( powerNums[i].multiply( BigInteger.valueOf(showTimes[i]) ) ); } return sum; } /** * 获取checkNum中每个数字各自的showTimes * @param checkNum * @return */ private static int[] getCheckShowTimes(BigInteger checkNum){ String s = checkNum.toString(); int[] checkShowTimes = new int[10]; for (int i = 0; i < s.length(); i++){ //取得一位数字的值，并且showTimes ++ int value = s.charAt(i) - '0'; checkShowTimes[value] += 1; } return checkShowTimes; } /** * 将0-9的showTimes换算成为水仙花和，并与最小值minNum比较， * 若小于最小值，则返回true，否则返回false。 * <p> * 只有showTimes形如 <br> * [0, 1, 2, 3, 4, 5, 6, 7, 8, 9] <br> * ------------------------------ <br> * [0, 0, k, n-k, 0, 0, 0, 0, 0, 0] <br> * 时才进行比较 * </p> * @param showTimes * @param minNum * @param n * @return */ private static boolean checkMinValue(int[] showTimes, BigInteger minNum, int n){ int maxShowedNum = 9; for(; showTimes[maxShowedNum] == 0; maxShowedNum --){} if(showTimes[maxShowedNum] + showTimes[maxShowedNum - 1] == n){ BigInteger sum = BigInteger.ZERO; sum = sum.add( powerNums[maxShowedNum].multiply( BigInteger.valueOf(showTimes[maxShowedNum]) ) ); sum = sum.add(powerNums[maxShowedNum - 1].multiply( BigInteger.valueOf(showTimes[maxShowedNum - 1]) )); return sum.compareTo(minNum) < 0; } return false; } }