解题报告_18.5.10_POJ_3006

素数筛选与Dirichlet素数查找

最新推荐文章于 2018-05-31 23:15:40 发布

转载最新推荐文章于 2018-05-31 23:15:40 发布 · 124 阅读

水题同时被 2 个专栏收录

20 篇文章

订阅专栏

POJ

13 篇文章

订阅专栏

本文介绍了一种使用筛法求解素数的方法，并在此基础上实现了寻找特定形式的Dirichlet素数的功能。通过C++代码示例展示了如何高效地找到前n个形式为a + kd (k ≥ 0)的素数。

难度渐渐上来了，不知道以后怎么办，这一道至少花去了我一个小时的时间。

主要是还筛法求素数

转载自：https://blog.youkuaiyun.com/lyy289065406/article/details/6642592

#include<memory.h>
#include<iostream>
#include<cmath>
#include<cstdio>
using namespace std;
const static int LEN = 1000000;
const static int SQRT_NUM = ceil(sqrt((double)LEN));
void findPrimes(bool* primes);
int findDirichletPrime(bool* primes, int a, int d, int n);
int  main() {
    freopen("in.txt","r",stdin);
    bool primes[LEN] ;
    findPrimes(primes);

    int a, d, n;
    while(cin >> a >> d >> n &&
    a != 0 && d != 0 && n != 0) {
        cout << findDirichletPrime(primes, a, d, n) << endl;
    }
}
void findPrimes(bool* primes) {
    memset(primes, true, sizeof(bool)*LEN);
    primes[0] = primes[1] = false;
    for(int i = 2; i <= SQRT_NUM; i++) {
        if(!primes[i]) continue;
        int multiple = 2;
        while(true) {
            int mNum = i * multiple;
            if(mNum > LEN) break;
            primes[mNum] = false;
            multiple++;
        }
    }
}
int findDirichletPrime(bool* primes, int a,int d, int n) {
    int ans = 0;
    for(int cnt = 0, dirichlet = a; cnt < n; dirichlet += d) {
        if(primes[dirichlet]) {
            ans = dirichlet;
            cnt++;
        }
    }
    return ans;
}