首頁 > 軟體

Java PTA 計算3到7位 水仙花數範例

2022-03-01 16:00:26

PTA計算3到7位 水仙花數

import java.util.Scanner;
public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        int n = in.nextInt();
        int min = (int) Math.pow(10, n - 1);
        int max = (int) Math.pow(10, n);
        int[] pow = new int[10];
        pow[1] = 1;
        for (int i = 2; i < pow.length; i++) {
            pow[i] = (int) Math.pow(i, n);
        }
        for (int i = min; i < max; i++) {
            int sum = 0;
            for (int j = 10; j < max; j *= 10) {
                sum += pow[i / j % 10];
            }
            sum += pow[i % 10];
            if (sum == i) {
                System.out.println(i);
            }
        }
    }
}

水仙花數的Java實現

春天是鮮花的季節,水仙花就是其中最迷人的代表,數學上有個水仙花數,他是這樣定義的:“水仙花數”是指一個三位數,它的各位數位的立方和等於其本身,比如:153=1^3+5^3+3^3。

要求輸出所有在m和n範圍內的水仙花數

import java.util.ArrayList;
import java.util.Scanner; 
public class Main {
    public static void main(String[] args) {
        Scanner input = new Scanner(System.in);
        int start = input.nextInt();
        int end = input.nextInt();// 系統輸入兩個整數
        ArrayList<Integer> out = flower(start, end);// 呼叫方法
        if (out.isEmpty()) {// 判斷是否為空,按格式輸出no。
            System.out.println("no");
        } else {
            for (int i = 0; i < out.size(); i++) {// 按格式輸出
                if (i != out.size() - 1) {// 最後一個不需要輸出空格
                    System.out.print(out.get(i) + " ");
                } else {
                    System.out.print(out.get(i));
                }
            }
        } 
    }
 
    public static ArrayList<Integer> flower(int startNumber, int endNumebr) { 
        ArrayList<Integer> flowerNumber = new ArrayList<Integer>();
        int j = 0;
        for (int i = startNumber; i < endNumebr; i++) {
            /*
             * 若滿足條件放入陣列 條件是分別取出3個位的數的3次方相加==原數
             */
            if ((i == Math.pow(Math.floor(i / 100), 3) + Math.pow(Math.floor(i % 100 / 10), 3)
                    + Math.pow(i % 100 % 10, 3)) && i >= startNumber && i <= endNumebr) { 
                flowerNumber.add(i); 
            }
        }
        return flowerNumber;// 返回陣列
    } 
}

程式碼還能優化

比如可以用陣列儲存0-9的三次冪,需要用到多少次冪的時候就用多少次冪,用陣列需要回圈判斷輸出,用hashmap直接對映比較好,減少三次冪運算。

這裡博主就直接使用math.pow()這個方法求冪了。

Math.pow(a,b)方法的使用是 a為底數,b為冪數。比如a的b次方。

以上為個人經驗,希望能給大家一個參考,也希望大家多多支援it145.com。


IT145.com E-mail:sddin#qq.com