判断素数 java版 PTA

本题的目标很简单，就是判断一个给定的正整数是否素数。

输入格式：
输入在第一行给出一个正整数N（≤ 10），随后N行，每行给出一个小于2
​31
​​ 的需要判断的正整数。

输出格式：
对每个需要判断的正整数，如果它是素数，则在一行中输出Yes，否则输出No。

输入样例：

2
11
111

输出样例：

Yes
No

代码实现

import java.util.Scanner;
//判断素数
public class Main{
    public static void main(String[] args) {
        Scanner scan = new Scanner(System.in);
        int n = scan.nextInt();
        int[] arr = new int[n];
        for (int i = 0; i < n; i++) {
            arr[i] = scan.nextInt();
        }
        for (int i : arr) {
            if (isPrime(i)){
                System.out.println("Yes");
            }else {
                System.out.println("No");
            }
        }
    }

    /**
     * 判断素数常规算法
     */
    public static boolean isPrime(int num) {
        boolean b = true;
        if (num == 1 || num == 0) {
            b = false;
        }
        //num必须开方，否则运行超时
        for (int i = 2; i <= Math.sqrt(num); i++) {
            if (num % i == 0) {
                b = false;
                break;
            }
        }
        return b;
    }
}