mirror of
https://github.com/TheAlgorithms/Java.git
synced 2025-07-06 17:29:31 +08:00
Add Collatz Conjecture (#3290)
This commit is contained in:
42
src/main/java/com/thealgorithms/maths/CollatzConjecture.java
Normal file
42
src/main/java/com/thealgorithms/maths/CollatzConjecture.java
Normal file
@ -0,0 +1,42 @@
|
|||||||
|
package com.thealgorithms.maths;
|
||||||
|
|
||||||
|
import java.util.ArrayList;
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
/**
|
||||||
|
* <a href="https://en.wikipedia.org/wiki/Collatz_conjecture">...</a>
|
||||||
|
*/
|
||||||
|
public class CollatzConjecture {
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Calculate the next number of the sequence.
|
||||||
|
*
|
||||||
|
* @param n current number of the sequence
|
||||||
|
* @return next number of the sequence
|
||||||
|
*/
|
||||||
|
public int nextNumber(final int n) {
|
||||||
|
if (n % 2 == 0) {
|
||||||
|
return n / 2;
|
||||||
|
}
|
||||||
|
return 3 * n + 1;
|
||||||
|
}
|
||||||
|
|
||||||
|
/**
|
||||||
|
* Calculate the Collatz sequence of any natural number.
|
||||||
|
*
|
||||||
|
* @param firstNumber starting number of the sequence
|
||||||
|
* @return sequence of the Collatz Conjecture
|
||||||
|
*/
|
||||||
|
public List<Integer> collatzConjecture(int firstNumber) {
|
||||||
|
if (firstNumber < 1) {
|
||||||
|
throw new IllegalArgumentException("Must be a natural number");
|
||||||
|
}
|
||||||
|
ArrayList<Integer> result = new ArrayList<>();
|
||||||
|
result.add(firstNumber);
|
||||||
|
while (firstNumber != 1) {
|
||||||
|
result.add(nextNumber(firstNumber));
|
||||||
|
firstNumber = nextNumber(firstNumber);
|
||||||
|
}
|
||||||
|
return result;
|
||||||
|
}
|
||||||
|
}
|
@ -0,0 +1,40 @@
|
|||||||
|
package com.thealgorithms.maths;
|
||||||
|
|
||||||
|
import org.junit.jupiter.api.BeforeAll;
|
||||||
|
import org.junit.jupiter.api.Test;
|
||||||
|
|
||||||
|
import java.util.List;
|
||||||
|
|
||||||
|
import static org.junit.jupiter.api.Assertions.*;
|
||||||
|
|
||||||
|
class CollatzConjectureTest {
|
||||||
|
|
||||||
|
static CollatzConjecture cConjecture;
|
||||||
|
|
||||||
|
@BeforeAll
|
||||||
|
static void setUp() {
|
||||||
|
cConjecture = new CollatzConjecture();
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
void nextNumberFromEvenNumber() {
|
||||||
|
assertEquals(25, cConjecture.nextNumber(50));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
void nextNumberFromOddNumber() {
|
||||||
|
assertEquals(154, cConjecture.nextNumber(51));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
void collatzConjecture() {
|
||||||
|
final List<Integer> expected = List.of(35, 106, 53, 160, 80, 40, 20, 10, 5, 16, 8, 4, 2, 1);
|
||||||
|
assertIterableEquals(expected, cConjecture.collatzConjecture(35));
|
||||||
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
void sequenceOfNotNaturalFirstNumber() {
|
||||||
|
assertThrows(IllegalArgumentException.class, () -> cConjecture.collatzConjecture(0));
|
||||||
|
assertThrows(IllegalArgumentException.class, () -> cConjecture.collatzConjecture(-1));
|
||||||
|
}
|
||||||
|
}
|
Reference in New Issue
Block a user