Added helpers for calculation primes.
This commit is contained in:
parent
69eb5748a9
commit
ef561e6e5a
|
@ -0,0 +1,108 @@
|
|||
#include <stdio.h>
|
||||
#include <stdlib.h>
|
||||
#include <stdbool.h>
|
||||
#include <math.h>
|
||||
|
||||
/*
|
||||
* unmemoized version of isPrime
|
||||
* runs in O(sqrt(n))
|
||||
*
|
||||
*/
|
||||
bool isPrime (int n) {
|
||||
if (n == 0 || n == 1)
|
||||
return false;
|
||||
|
||||
int sqrtn = (int) sqrt (n);
|
||||
for (int i = 2; i <= sqrtn; i++) {
|
||||
if (n % i == 0)
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/*
|
||||
* memoized version of isPrime
|
||||
* primes: list of primes < n
|
||||
* runs in O(sqrt(n)/ln(n))
|
||||
*
|
||||
*/
|
||||
bool isPrimeMem (int n, int* primes) {
|
||||
if (n == 0 || n == 1)
|
||||
return false;
|
||||
if (n == 2 || n == 3)
|
||||
return true;
|
||||
|
||||
int sqrtn = (int) sqrt (n);
|
||||
int i = 0;
|
||||
while (primes[i] <= sqrtn) {
|
||||
if (n % primes[i++] == 0)
|
||||
return false;
|
||||
}
|
||||
return true;
|
||||
}
|
||||
|
||||
/*
|
||||
* calculates primes <= n using dynamic programming
|
||||
*
|
||||
* n: non-negative int
|
||||
* primes_ptr: pointer to array of primes <= n
|
||||
* primesTable_ptr: pointer to array of size n+1
|
||||
* where the 0-based ith element is true if i is prime
|
||||
* returns: number of primes in *primes_ptr
|
||||
*
|
||||
* N.B. numOfPrimesUpper is an upper-bound of
|
||||
* the prime-counting function, given by
|
||||
* n/ln(n) * (1 + 3/2 * 1/ln(n))
|
||||
*
|
||||
* runs in O(n * sqrt(n)/ln(n)) (not a tight bound)
|
||||
*
|
||||
*/
|
||||
|
||||
int listOfPrimes (int n, int** primes_ptr, bool** primesTable_ptr) {
|
||||
*primesTable_ptr = calloc (n + 1, sizeof (bool));
|
||||
|
||||
if (n == 0 || n == 1) {
|
||||
*primes_ptr = malloc (0);
|
||||
return 0;
|
||||
}
|
||||
|
||||
int numOfPrimesUpper = (int) (n/log(n) *
|
||||
(1 + 3/2 * 1/log(n)));
|
||||
*primes_ptr = malloc (sizeof (int) * numOfPrimesUpper);
|
||||
int numOfPrimes = 0;
|
||||
for (int i = 0; i <= n; i++) {
|
||||
if (isPrimeMem (i, *primes_ptr)) {
|
||||
(*primes_ptr)[numOfPrimes++] = i;
|
||||
(*primesTable_ptr)[i] = true;
|
||||
}
|
||||
}
|
||||
if (numOfPrimes < numOfPrimesUpper)
|
||||
*primes_ptr = realloc (*primes_ptr,
|
||||
sizeof (int) * numOfPrimes);
|
||||
|
||||
return numOfPrimes;
|
||||
}
|
||||
|
||||
/*
|
||||
* calculates primes <= n are prime using the sieve of Eratosthenes
|
||||
*
|
||||
* runs in O(n ln(ln(n))) apparently
|
||||
*
|
||||
*/
|
||||
void sieveOfEratosthenes (int n, bool** primesTable_ptr) {
|
||||
*primesTable_ptr = malloc (sizeof (bool) * (n + 1));
|
||||
for (int i = 0; i <= n; i++)
|
||||
(*primesTable_ptr)[i] = true;
|
||||
|
||||
(*primesTable_ptr)[0] = false;
|
||||
(*primesTable_ptr)[1] = false;
|
||||
|
||||
int sqrtn = (int) sqrt (n);
|
||||
for (int i = 0; i <= sqrtn; i++) {
|
||||
if ((*primesTable_ptr)[i] == true) {
|
||||
for (int j = i*2; j <= n; j += i) {
|
||||
(*primesTable_ptr)[j] = false;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
Loading…
Reference in New Issue