项目欧拉#10如何预先计算素数之和?

时间:2015-06-14 15:16:50

标签: c++ c++11 primes sieve-of-eratosthenes

以下是问题:

低于10的素数之和为2 + 3 + 5 + 7 = 17.

求出不大于给定N的所有素数之和。

输入格式: 第一行包含整数T,即测试用例的数量。 下一个T行将包含一个整数N。

输出格式: 在单独的行中打印与每个测试用例对应的值。

限制条件: 1≤T≤104 1≤N≤106

https://www.hackerrank.com/contests/projecteuler/challenges/euler010 这是问题的链接。

所以,我试图用Eratosthenes筛来解决这个问题。 我预先计算了10 ^ 6以下的所有素数,这是N的给定限制。

7个测试用例中有6个被接受,但最后一个测试用例给出了Timeout(TLE)。

我阅读了讨论论坛并在那里说,为了解决这个问题,我们还需要预先计算素数的总和。

所以,我尝试制作一个长期整数的数组,并尝试将所有总和存储在其中。但这给了我一个分段错误。

那么,我怎么能预先计算素数的总和呢?

这是我的代码:

#include "header.h"             //MAX is defined to be 1000000

bool sieve[MAX + 1];        // false = prime, true = composite

int main(void){
    //0 and 1 are not primes
    sieve[0] = sieve[1] = true;
    //input limiting value
    int n = MAX;
    //cross out even numbers
    for(int i = 4; i <= n; i += 2){
        sieve[i] = true;
    }
    //use sieve of eratosthenes
    for(int i = 3; i <= static_cast<int>(sqrt(n)); i += 2){     
        if(sieve[i] == false){                          
            for(int j = i * i; j <= n; j += i)              
                sieve[j] = true;
        }
    }
    long long p, ans = 0;
    int t;
    std::cin >> t;
    while(t--){
        std::cin >> p;
        for(int i = 0; i <= p; ++i)
            if(sieve[i] == false)
                ans += i;
        std::cout << ans << std::endl;
        ans = 0;
    }
    return 0;
}

1 个答案:

答案 0 :(得分:3)

给定一个素数数组prime[N],素数的预计算总和可以在一个for循环中完成,如下所示:

int sum[N];
sum[0] = primes[0];
for (int i = 1 ; i < N ; i++) {
    sum[i] = prime[i]+sum[i-1];
}

您可以将此数组与primes[]一起使用,在primes上运行二进制搜索,如果搜索的数字是素数,则在同一位置选择sum,或者如果数字不是素数,则为先前的位置。