蓝桥杯-找素数【筛选法】

  算法提高 找素数  

时间限制：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;
}