c++自定义函数判断素数

Question

输入一个正整数n，求第n小的质数，第n小就是从2开始的第n个。输入10输出 29。

Answer 1

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

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

int findNthPrime(int n) {
    int count = 0, num = 1;
    while (count < n) {
        num++;
        if (isPrime(num)) count++;
    }
    return num;
}

int main() {
    int n;
    cin >> n;
    cout << findNthPrime(n) << endl;
    return 0;
}