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.
一开始想的有点复杂了，多花费了一些时间。