POJ3006-Dirichlet's Theorem on Arithmetic Progressions

全解题报告索引目录 -> 【北大ACM – POJ试题分类】

转载请注明出处：http://exp-blog.com

-------------------------------------------------------------------------

 

题目大意

狄利克雷基于等差数列的算法原理

设一个等差数列，首元素为a，公差为b

现在要求输入a,b,n ，要求找出属于该等差数列中的第n个素数并输出

 

 

/*
	Author:     Exp
	Date:       2017-11-30
	Code:       POJ 3006
	Problem:    Dirichlet's Theorem on Arithmetic Progressions
	URL:		http://poj.org/problem?id=3006
*/

/*
	题意分析：
	 纯粹引用了狄利克雷数列，但其实没用到狄利克雷定理的问题。
	 狄利克雷数列即形如 a+nd 的等差数列，其中a、d互质（公约数只有1的两个自然数），n=1,2,3,......
	 已知在狄利克雷数列中，会出现无数个素数（这些质数模d同余a，但本题没用到这个特性）

	 现给定a和d，通过构造其狄利克雷数列，求数列中的第n个素数。
	 其中 a∈[1,9307], d∈[1,346], n∈[1,210]


	解题思路：
	 这题的关键不是狄利克雷数列，还是求素数。
	 素数的范围可以通过 a、d、n 的最大值圈定（根据给定的数列参数最大值，约为100W以内的素数）
	 有了素数表之后就是暴力美学的工作了

	 ① 用筛法求取题目给定最大范围内的所有素数
	 ② 根据a、d构造狄利克雷数列，通过素数表找到其中第n个素数
	 ③ 遍历数列时不要用 a+nd 公式，去掉乘法， 只用加法循环 a += d 即可

*/

#include <memory.h>
#include <math.h>
#include <iostream>
using namespace std;


const static int LEN = 1000000;							// 自然数数组长度(求解素数范围)
const static int SQRT_NUM = ceil(sqrt((double) LEN));	// 根据合数定理得到的质因数范围

/* 
 * 使用筛法找出自然数范围内的所有素数
 * @param primes 素数表
 */
void findPrimes(bool* primes);

/* 
 * 获取狄利克雷数列中第n个素数
 * @param primes 素数集
 * @param a 狄利克雷数列参数 a∈[1,9307]
 * @param d 狄利克雷数列参数 b∈[1,346]
 * @param n 期望得到素数序次 n∈[1,210]
 * return 期望得到的第n个素数
 */
int getDirichletPrime(bool* primes, int a, int d, int n);

int main(void) {
	bool primes[LEN];		// 素数集, 标记每个自然数是否为素数
	findPrimes(primes);		// 找出范围内所有素数

	int a, d, n;
	while(cin >> a >> d >> n && 
		a != 0 && d != 0 && n != 0) {	// 注意POJ测试用例中有个陷阱: a非质数,d=0,n>0, 若不单独约束d=0这个条件会因死循环导致TLE
		int prime = getDirichletPrime(primes, a, d, n);
		cout << prime << endl;
	}
	return 0;
}


void findPrimes(bool* primes) {
	memset(primes, true, sizeof(bool) * LEN);	// 注意memset是按字节覆写内存的
	primes[0] = primes[1] = false;

	for(int i = 2; i <= SQRT_NUM; i++) {
		if(primes[i] == false) {
			continue;
		}

		// 筛掉最小素数的所有倍数
		int multiple = 2;	// i的倍率（因不包括自身, 从2倍开始）	
		while(true) {
			int mNum = i * multiple;	// i的倍数
			if(mNum >= LEN) {
				break;
			}
			primes[mNum] = false;
			multiple++;
		}
	}
}

int getDirichletPrime(bool* primes, int a, int d, int n) {
	int prime = 0;		// 第n个素数
	for(int cnt = 0, dirichlet = a; cnt < n; dirichlet += d) {
		if(primes[dirichlet] == true) {
			prime = dirichlet;
			cnt++;
		}
	}
	return prime;
}