算法提高 找素数
时间限制:1.0s 内存限制:256.0MB
问题描述
给定区间[L, R] , 请计算区间中素数的个数。
输入格式
两个数L和R。
输出格式
一行,区间中素数的个数。
样例输入
2 11
样例输出
5
数据规模和约定
2 <= L <= R <= 2147483647 R-L <= 1000000
一般的方法会超时。。
数组开小了,也会超时。
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cmath>
using namespace std;
const int MAX_L = 9999999;
const int MAX_SQRT_B = 9999999;
typedef long long ll;
bool is_prime[MAX_L], is_prime_small[MAX_SQRT_B];
int sum = 0;
void segment_sieve(ll a, ll b) {
for (int i = 0; (ll)i * i < b; i++) is_prime_small[i] = true;
for (int i = 0; i < b - a; i++) is_prime[i] = true;
for (int i = 2; (ll)i * i < b; i++) {
if (is_prime_small[i]) {
for (int j = 2*i; (ll)j*j < b; j += i) is_prime_small[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;
cin >> a >> b;
segment_sieve(a, b+1);
for (ll i = a; i <= b; i++) {
if (is_prime[i-a]) sum++;
}
printf("%d\n", sum);
return 0;
}