蓝桥杯 - 找素数

这道题用普通的方法会超时

首先用筛素数的方法，把素数筛除来，然后在这其中，维护一个数组，记录这个数是否是素数

#include
#include
#include
#include
#include

using namespace std;

typedef long long ll;
const int MAXN = 9999999;
int sum = 0;
bool is_prime[MAXN], vis[MAXN];

void getprime(ll a, ll b) {
    int m = sqrt(b+0.5);
    memset(vis, true, sizeof(vis));
    for(int i = 0; i < b - a; i++)
        is_prime[i] = true;
    for(int i = 2; (ll)i * i < b; i++) {
        if(vis[i]) {
            for(int j = 2*i; (ll)j*j < b; j += i) vis[j] = false;
            for(ll j = max(2LL, (a+i-1)/i)*i; j < b; j += i) {
                is_prime[j-a] = false;
            }
        }
    }
    return;
}
int main() {
    ll a, b;
    scanf("%lld %lld", &a, &b);
    getprime(a, b+1);
    for(ll i = a; i <= b; i++) {
        if(is_prime[i-a]) sum++;
    }
    printf("%d\n", sum);
    return 0;
}