L1-028. 判斷素數（Java）PAT團體程序設計天梯賽-練習集

本題的目標很簡單，就是判斷一個給定的正整數是否素數。

輸入格式：

輸入在第一行給出一個正整數N（<=10），隨後N行，每行給出一個小於231的需要判斷的正整數。

輸出格式：

對每個需要判斷的正整數，如果它是素數，則在一行中輸出“Yes”，否則輸出“No”。

輸入樣例：

2
11
111

輸出樣例：

Yes
No

---

import java.util.Scanner;

public class Main {

    // 判斷是否是素數
    public static boolean isPrime(int num) {
        if (num == 1 || num == 0)
        {
            return false;
        } 

        for (int i  = 2; i <= Math.sqrt(num); i++)
        {
            if (num % i == 0)
            {
                return false;
            }       
        }

        return true;
    }

    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        int N = scanner.nextInt();

        while (N-- > 0)
        {
            int num = scanner.nextInt();
            System.out.println(isPrime(num) ? "Yes" : "No");
         } 
    }
}