c++编写一个函数，接受一个正整数 n，计算并返回第 n 个素数。

c++编写一个函数，接受一个正整数 n，计算并返回第 n 个素数。

#include <iostream>
#include <vector>

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 nthPrime(int n) {
    if (n <= 0) {
        return 0; // 返回 0 表示无效输入
    }

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

int main() {
    int n = 5;
    int result = nthPrime(n);
    std::cout << "The " << n << "th prime number is: " << result << std::endl;
    return 0;
}