PTA L1-028 判断素数 (C++实现)

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

输入格式：

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

输出格式：

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

输入样例：

2
11
111

输出样例：

Yes
No

代码实现：

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

bool isPrime(int num) {
    if (num < 2)
        return false;
    for (int i = 2; i <= sqrt(num); i++) {
        if (num % i == 0)
            return false;
    }
    return true;
}

int main() {
    int count;
    int a[10000];
    cin >> count;
    for (int i = 0; i < count; i++) {
        cin >> a[i];
    }
    for (int i = 0; i < count; i++) {
        int num = a[i];
        if (isPrime(num)) {
            cout << "Yes" << endl;
        } else {
            cout << "No" << endl;
        }
    }
    return 0;
}

函数中的循环部分可以这样理解：

对于一个数如果存在因子 那么他至少可以表示成两个数相乘的状态 如：N = a * b，其中a <= b，如果a大于平方根N，那么b必然小于平方根N，否则a * b就会大于N。所以，我们只需在2到平方根N之间检查因子a是否存在就足够了。