POJ 3006 Dirichlet's Theorem on Arithmetic Progressions
Time Limit: 1000MS | Memory Limit: 65536K | |
Total Submissions: 11334 | Accepted: 5693 |
Description
If a and d are relatively prime positive integers, the arithmetic sequence beginning with a and increasing by d, i.e., a, a + d, a + 2d, a + 3d, a + 4d, ..., contains infinitely many prime numbers. This fact is known as Dirichlet's Theorem on Arithmetic Progressions, which had been conjectured by Johann Carl Friedrich Gauss (1777 - 1855) and was proved by Johann Peter Gustav Lejeune Dirichlet (1805 - 1859) in 1837.
For example, the arithmetic sequence beginning with 2 and increasing by 3, i.e.,
2, 5, 8, 11, 14, 17, 20, 23, 26, 29, 32, 35, 38, 41, 44, 47, 50, 53, 56, 59, 62, 65, 68, 71, 74, 77, 80, 83, 86, 89, 92, 95, 98, ... ,
contains infinitely many prime numbers
2, 5, 11, 17, 23, 29, 41, 47, 53, 59, 71, 83, 89, ... .
Your mission, should you decide to accept it, is to write a program to find the nth prime number in this arithmetic sequence for given positive integers a, d, and n.
Input
The input is a sequence of datasets. A dataset is a line containing three positive integers a, d, and n separated by a space. a and d are relatively prime. You may assume a <= 9307, d <= 346, and n <= 210.
The end of the input is indicated by a line containing three zeros separated by a space. It is not a dataset.
Output
The output should be composed of as many lines as the number of the input datasets. Each line should contain a single integer and should never contain extra characters.
The output integer corresponding to a dataset a, d, n should be the nth prime number among those contained in the arithmetic sequence beginning with a and increasing by d.
FYI, it is known that the result is always less than 106 (one million) under this input condition.
Sample Input
367 186 151
179 10 203
271 37 39
103 230 1
27 104 185
253 50 85
1 1 1
9075 337 210
307 24 79
331 221 177
259 170 40
269 58 102
0 0 0
Sample Output
92809
6709
12037
103
93523
14503
2
899429
5107
412717
22699
25673
Source
1 #include <iostream>
2 #include <cstdlib>
3 #include <algorithm>
4 #include <cstring>
5 #include <cstdio>
6
7 #define DATASIZE 1000000
8
9 using namespace std;
10
11 bool flag[DATASIZE];
12 int prime[DATASIZE];
13
14 int generate_prime(void)
15 {
16 memset(flag,true,sizeof(bool)*DATASIZE);
17 int current = 0;
18
19 for(int i = 2;i*i <= DATASIZE;++i)
20 if(flag[i])
21 for(int j = 2*i;j <= DATASIZE;j += i)
22 flag[j] = false;
23 for(int i = 2;i != DATASIZE;++i)
24 if(flag[i])
25 prime[current++] = i;
26 return current-1;
27 }
28
29 int main(void)
30 {
31 int a,d,n,result[250];
32 int primecount = generate_prime();
33
34 while(cin >> a >> d >> n && a+d+n)
35 {
36 memset(result,0,sizeof(int)*250);
37 int count = 0;
38 for(int i = 0;i != primecount;++i)
39 {
40 if(prime[i] < a)
41 continue;
42 else if((prime[i]-a)%d == 0)
43 {
44 count++;
45 if(count == n)
46 {
47 printf("%d\n",prime[i]);
48 break;
49 }
50 }
51 }
52 }
53 return 0;
54 }