PAT (Basic Level) 1013 数素数 (20 point(s))

#include <bits/stdc++.h>
#define MANX 1000001
using namespace std;

// 判断素数
bool isPrime(int n){
    for(int k = 2; k <= (int)sqrt(n); k++)
        if(n % k == 0)
            return false;
    return true;
}

int main() {
    int m, n, cnt = 0, prime[MANX];
    cin >> m >> n;
    // 获取素数
    for(int i = 2; cnt != n; i++){
        // 第n个素数停止
        if(isPrime(i) == true)
            prime[++cnt] = i;
    }
    // 将m转换成从1开始的序号 比如m=5 即i=i-m+1 = 1
    for(int i = m; i <= n; i++){
        // 首个无空格 个位数为一
        if((i - m + 1) % 10 != 1) cout << ' ';
        cout << prime[i];
        // 十个输出换一行
        if((i - m + 1) % 10 == 0) cout << endl;
    }
}

#include <bits/stdc++.h>
#define MANX 1000001
using namespace std;

int main() {
	int M, N, pNum = 0,  pVis[MANX]{false}, prime[MANX];
	cin >> M >> N;
    // 埃氏筛法
	for (int i = 2; pNum != N; ++i) {
        if(pVis[i] == false){
            prime[++pNum] = i;
            for(int j = i + i; j < MANX; j += i){
                pVis[j] = true;
            }
        }
	}
	for (int i = M; i <= pNum; ++i) {
		if ((i - M + 1) % 10 != 1) cout << " ";
		cout << prime[i];
		if ((i - M + 1) % 10 == 0) cout << endl;
	}
}

k <= (int)sqrt(n)

判断素数的判断条件。设被判断的数为 n 除数为 i ，则根据算法笔记里面，k * n / k = n = √n * √n则满足 k 和 n / k 一个小于 √n 一个大于 √n。所以 k <= √n 就可以得到此处的循环条件，这比 k 遍历到 n 要快不少。

---

第二种方法，因为题目最大为第10^4个素数，而MANX为正整数需要远大于10^4。而如果定太大，并且数组放在main函数里面，可能会导致函数栈爆掉。而如果MANX定小了，那么可能会由于没数到答案要求的第N个素数错误。

在牛客网上面看数据，有个数据集数到了十万，所以这里就定义为百万了。如果不这么麻烦，就第一种方法也可以，我这里只是单纯试一下另一种方法。

参考代码