PAT BASIC LEVEL 1013. 数素数 (20)

1013. 数素数 (20)

令Pi表示第i个素数。现任给两个正整数M <= N <= 104，请输出PM到PN的所有素数。

输入格式：

输入在一行中给出M和N，其间以空格分隔。

输出格式：

输出从PM到PN的所有素数，每10个数字占1行，其间以空格分隔，但行末不得有多余空格。

输入样例：

5 27

输出样例：

11 13 17 19 23 29 31 37 41 43

47 53 59 61 67 71 73 79 83 89

97 101 103

Answer:

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

bool is_prime(int n) {
double max = sqrt(n);
for(int i = 2; i <= max; i ++)
if(n % i == 0)
return false;
return true;
}

void count_prime(int start, int end, int* result) {
for(int i = 0, n = 2;i <= end; n ++)
if(is_prime(n) && ++i >= start)
*result++ = n;
}

int main() {
int start, end, count;
cin >> start >> end;
int result[end - start + 1];
count_prime(start, end, result);
for(int i = 0;i < end - start + 1;i ++) {
if(i % 10 == 0)
cout << result[i];
else if(i % 10 == 9)
cout << ' ' << result[i] << endl;
else
cout << ' ' << result[i];
}
}

PS.

一开始想的有点复杂了，多花费了一些时间。