merge: optamization,documantation,return Array[bool] => Array[primes only], test case (#841)

This commit is contained in:
YATIN KATHURIA
2021-11-20 17:15:36 +05:30
committed by GitHub
parent 93e57b0de5
commit 02a4cee5c2
2 changed files with 41 additions and 11 deletions

View File

@ -1,21 +1,28 @@
/**
* @function SieveOfEratosthenes
* @description Calculates prime numbers till input number n
* @param {Number} n - The input integer
* @return {Number[]} List of Primes till n.
* @see [Sieve_of_Eratosthenes](https://www.geeksforgeeks.org/sieve-of-eratosthenes/)
*/
function sieveOfEratosthenes (n) {
/*
* Calculates prime numbers till a number n
* :param n: Number up to which to calculate primes
* :return: A boolean list containing only primes
*/
const primes = new Array(n + 1)
primes.fill(true) // set all as true initially
if (n <= 1) return []
const primes = new Array(n + 1).fill(true) // set all as true initially
primes[0] = primes[1] = false // Handling case for 0 and 1
const sqrtn = Math.ceil(Math.sqrt(n))
for (let i = 2; i <= sqrtn; i++) {
for (let i = 2; i * i <= n; i++) {
if (primes[i]) {
for (let j = 2 * i; j <= n; j += i) {
for (let j = i * i; j <= n; j += i) {
primes[j] = false
}
}
}
return primes
return primes.reduce((result, isPrime, index) => {
if (isPrime) {
result.push(index)
}
return result
}, [])
}
// Example