题解 | 素数判断

import java.util.Scanner;

// 注意类名必须为 Main, 不要有任何 package xxx 信息
public class Main {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);

        // 读取测试用例数量
        int T = scanner.nextInt();

        // 处理每个测试用例
        for (int i = 0; i < T; i++) {
            int n = scanner.nextInt();
            if (isPrime(n)) {
                System.out.println("Yes");
            } else {
                System.out.println("No");
            }
        }

        scanner.close();
    }

    // 判断是否为素数的方法
    public static boolean isPrime(int n) {
        // 小于等于1的数不是素数
        if (n <= 1) {
            return false;
        }

        // 2是素数
        if (n == 2) {
            return true;
        }

        // 偶数(除了2)不是素数
        if (n % 2 == 0) {
            return false;
        }

        // 只需检查到sqrt(n)
        for (int i = 3; i * i <= n; i += 2) {
            if (n % i == 0) {
                return false;
            }
        }

        return true;
    }
}