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");
         } 
    }
}