题解 | 素数判断

#include <iostream>
#include <cmath>
using namespace std;

void sushu(int n) {
    // 是否素数
    if (n == 1) {
        cout << "No";
    }
    else if (n == 2) {
        cout << "Yes";
    }
    else {
        int judge = 0;
        for (int i = 2; i <= (sqrt(n) + 1); i++) {
            if (n % i == 0) {
                judge = 1;
                break;
            }
        }
        if (judge == 0) {
            cout << "Yes";
        }
        else {
            cout << "No";
        }
    }
}

int main() {
    int T, n;
    cin >> T;
    for (int i = 1; i <= T; i++) {
        cin >> n;
        sushu(n);
        cout << endl;
    }
    return 0;
}