mirror of
https://github.com/TheAlgorithms/Java.git
synced 2025-07-10 21:43:15 +08:00
tests: add tests for SieveOfEratosthenes
(#4217)
* tests: add tests for SieveOfEratosthenes * feat: throw for inputs <= 0 * refactor: simplify logic in SieveOfEratosthenes.findPrimesTill * refactor: make SieveOfEratosthenes a utility class * docs: update docs, fix typo
This commit is contained in:
@ -3,55 +3,63 @@ package com.thealgorithms.others;
|
|||||||
import java.util.Arrays;
|
import java.util.Arrays;
|
||||||
|
|
||||||
/**
|
/**
|
||||||
* Sieve of Eratosthenes is an ancient algorithm for finding all prime numbers
|
* @brief utility class implementing <a href="https://en.wikipedia.org/wiki/Sieve_of_Eratosthenes">Sieve of Eratosthenes</a>
|
||||||
* up to any given limit.
|
|
||||||
*
|
|
||||||
* @see <a href="https://en.wikipedia.org/wiki/Sieve_of_Eratosthenes">Wiki</a>
|
|
||||||
*/
|
*/
|
||||||
public class SieveOfEratosthenes {
|
final public class SieveOfEratosthenes {
|
||||||
|
private SieveOfEratosthenes() {
|
||||||
|
}
|
||||||
|
|
||||||
/**
|
private static void checkInput(int n) {
|
||||||
* Finds all prime numbers till n.
|
if (n <= 0) {
|
||||||
*
|
throw new IllegalArgumentException("n must be positive.");
|
||||||
* @param n The number till which we have to check for primes. Should be more than 1.
|
}
|
||||||
* @return Array of all prime numbers between 0 to n.
|
}
|
||||||
*/
|
|
||||||
public static int[] findPrimesTill(int n) {
|
private static Type[] sievePrimesTill(int n) {
|
||||||
Type[] numbers = new Type[n + 1];
|
checkInput(n);
|
||||||
Arrays.fill(numbers, Type.PRIME);
|
Type[] isPrimeArray = new Type[n + 1];
|
||||||
numbers[0] = numbers[1] = Type.NOT_PRIME;
|
Arrays.fill(isPrimeArray, Type.PRIME);
|
||||||
|
isPrimeArray[0] = isPrimeArray[1] = Type.NOT_PRIME;
|
||||||
|
|
||||||
double cap = Math.sqrt(n);
|
double cap = Math.sqrt(n);
|
||||||
for (int i = 2; i <= cap; i++) {
|
for (int i = 2; i <= cap; i++) {
|
||||||
if (numbers[i] == Type.PRIME) {
|
if (isPrimeArray[i] == Type.PRIME) {
|
||||||
for (int j = 2; i * j <= n; j++) {
|
for (int j = 2; i * j <= n; j++) {
|
||||||
numbers[i * j] = Type.NOT_PRIME;
|
isPrimeArray[i * j] = Type.NOT_PRIME;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
return isPrimeArray;
|
||||||
|
}
|
||||||
|
|
||||||
int primesCount = (int) Arrays.stream(numbers).filter(element -> element == Type.PRIME).count();
|
private static int countPrimes(Type[] isPrimeArray) {
|
||||||
int[] primes = new int[primesCount];
|
return (int) Arrays.stream(isPrimeArray).filter(element -> element == Type.PRIME).count();
|
||||||
|
}
|
||||||
|
|
||||||
|
private static int[] extractPrimes(Type[] isPrimeArray) {
|
||||||
|
int numberOfPrimes = countPrimes(isPrimeArray);
|
||||||
|
int[] primes = new int[numberOfPrimes];
|
||||||
int primeIndex = 0;
|
int primeIndex = 0;
|
||||||
for (int i = 0; i < n + 1; i++) {
|
for (int curNumber = 0; curNumber < isPrimeArray.length; ++curNumber) {
|
||||||
if (numbers[i] == Type.PRIME) {
|
if (isPrimeArray[curNumber] == Type.PRIME) {
|
||||||
primes[primeIndex++] = i;
|
primes[primeIndex++] = curNumber;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
return primes;
|
||||||
|
}
|
||||||
|
|
||||||
return primes;
|
/**
|
||||||
|
* @brief finds all of the prime numbers up to the given upper (inclusive) limit
|
||||||
|
* @param n upper (inclusive) limit
|
||||||
|
* @exception IllegalArgumentException n is non-positive
|
||||||
|
* @return the array of all primes up to the given number (inclusive)
|
||||||
|
*/
|
||||||
|
public static int[] findPrimesTill(int n) {
|
||||||
|
return extractPrimes(sievePrimesTill(n));
|
||||||
}
|
}
|
||||||
|
|
||||||
private enum Type {
|
private enum Type {
|
||||||
PRIME,
|
PRIME,
|
||||||
NOT_PRIME,
|
NOT_PRIME,
|
||||||
}
|
}
|
||||||
|
|
||||||
public static void main(String[] args) {
|
|
||||||
int n = 100;
|
|
||||||
System.out.println("Searching for all primes from zero to " + n);
|
|
||||||
int[] primes = findPrimesTill(n);
|
|
||||||
System.out.println("Found: " + Arrays.toString(primes));
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
@ -0,0 +1,46 @@
|
|||||||
|
package com.thealgorithms.others;
|
||||||
|
|
||||||
|
import static org.junit.jupiter.api.Assertions.assertArrayEquals;
|
||||||
|
import static org.junit.jupiter.api.Assertions.assertThrows;
|
||||||
|
|
||||||
|
import org.junit.jupiter.api.Test;
|
||||||
|
|
||||||
|
class SieveOfEratosthenesTest {
|
||||||
|
@Test
|
||||||
|
public void testfFindPrimesTill1() {
|
||||||
|
assertArrayEquals(new int[] {}, SieveOfEratosthenes.findPrimesTill(1));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testfFindPrimesTill2() {
|
||||||
|
assertArrayEquals(new int[] {2}, SieveOfEratosthenes.findPrimesTill(2));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testfFindPrimesTill4() {
|
||||||
|
var primesTill4 = new int[] {2, 3};
|
||||||
|
assertArrayEquals(primesTill4, SieveOfEratosthenes.findPrimesTill(3));
|
||||||
|
assertArrayEquals(primesTill4, SieveOfEratosthenes.findPrimesTill(4));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testfFindPrimesTill40() {
|
||||||
|
var primesTill40 = new int[] {2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37};
|
||||||
|
assertArrayEquals(primesTill40, SieveOfEratosthenes.findPrimesTill(37));
|
||||||
|
assertArrayEquals(primesTill40, SieveOfEratosthenes.findPrimesTill(38));
|
||||||
|
assertArrayEquals(primesTill40, SieveOfEratosthenes.findPrimesTill(39));
|
||||||
|
assertArrayEquals(primesTill40, SieveOfEratosthenes.findPrimesTill(40));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testfFindPrimesTill240() {
|
||||||
|
var primesTill240 = new int[] {2, 3, 5, 7, 11, 13, 17, 19, 23, 29, 31, 37, 41, 43, 47, 53, 59, 61, 67, 71, 73, 79, 83, 89, 97, 101, 103, 107, 109, 113, 127, 131, 137, 139, 149, 151, 157, 163, 167, 173, 179, 181, 191, 193, 197, 199, 211, 223, 227, 229, 233, 239};
|
||||||
|
assertArrayEquals(primesTill240, SieveOfEratosthenes.findPrimesTill(239));
|
||||||
|
assertArrayEquals(primesTill240, SieveOfEratosthenes.findPrimesTill(240));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
public void testFindPrimesTillThrowsExceptionForNonPositiveInput() {
|
||||||
|
assertThrows(IllegalArgumentException.class, () -> SieveOfEratosthenes.findPrimesTill(0));
|
||||||
|
}
|
||||||
|
}
|
Reference in New Issue
Block a user