To add Suffix array algorithm implementation (#6530)

* adding Suffix Array Implementation

* adding test for Suffix Array

* patch: lint fix in SuffixArray.java

* patch: lint fix in SuffixArrayTest.java

* fix: private constructor update

* patch: lint fix in suffix array impl

* fix: lint issue line blankspace removal

* fix: checkstype issue

* fix: lint format issue

* Update DIRECTORY.md
This commit is contained in:
Nithin U
2025-09-17 21:26:10 +05:30
committed by GitHub
parent c370101151
commit 85a2df46ad
3 changed files with 106 additions and 0 deletions

View File

@@ -744,6 +744,7 @@
- 📄 [Rotation](src/main/java/com/thealgorithms/strings/Rotation.java)
- 📄 [StringCompression](src/main/java/com/thealgorithms/strings/StringCompression.java)
- 📄 [StringMatchFiniteAutomata](src/main/java/com/thealgorithms/strings/StringMatchFiniteAutomata.java)
- 📄 [SuffixArray](src/main/java/com/thealgorithms/strings/SuffixArray.java)
- 📄 [Upper](src/main/java/com/thealgorithms/strings/Upper.java)
- 📄 [ValidParentheses](src/main/java/com/thealgorithms/strings/ValidParentheses.java)
- 📄 [WordLadder](src/main/java/com/thealgorithms/strings/WordLadder.java)
@@ -1413,6 +1414,7 @@
- 📄 [RotationTest](src/test/java/com/thealgorithms/strings/RotationTest.java)
- 📄 [StringCompressionTest](src/test/java/com/thealgorithms/strings/StringCompressionTest.java)
- 📄 [StringMatchFiniteAutomataTest](src/test/java/com/thealgorithms/strings/StringMatchFiniteAutomataTest.java)
- 📄 [SuffixArrayTest](src/test/java/com/thealgorithms/strings/SuffixArrayTest.java)
- 📄 [UpperTest](src/test/java/com/thealgorithms/strings/UpperTest.java)
- 📄 [ValidParenthesesTest](src/test/java/com/thealgorithms/strings/ValidParenthesesTest.java)
- 📄 [WordLadderTest](src/test/java/com/thealgorithms/strings/WordLadderTest.java)

View File

@@ -0,0 +1,60 @@
package com.thealgorithms.strings;
import java.util.Arrays;
/**
* Suffix Array implementation in Java.
* Builds an array of indices that represent all suffixes of a string in sorted order.
* Wikipedia Reference: https://en.wikipedia.org/wiki/Suffix_array
* Author: Nithin U.
* Github: https://github.com/NithinU2802
*/
public final class SuffixArray {
private SuffixArray() {
}
public static int[] buildSuffixArray(String text) {
int n = text.length();
Integer[] suffixArray = new Integer[n];
int[] rank = new int[n];
int[] tempRank = new int[n];
// Initial ranking based on characters
for (int i = 0; i < n; i++) {
suffixArray[i] = i;
rank[i] = text.charAt(i);
}
for (int k = 1; k < n; k *= 2) {
final int step = k;
// Comparator: first by rank, then by rank + step
Arrays.sort(suffixArray, (a, b) -> {
if (rank[a] != rank[b]) {
return Integer.compare(rank[a], rank[b]);
}
int ra = (a + step < n) ? rank[a + step] : -1;
int rb = (b + step < n) ? rank[b + step] : -1;
return Integer.compare(ra, rb);
});
// Re-rank
tempRank[suffixArray[0]] = 0;
for (int i = 1; i < n; i++) {
int prev = suffixArray[i - 1];
int curr = suffixArray[i];
boolean sameRank = rank[prev] == rank[curr] && ((prev + step < n ? rank[prev + step] : -1) == (curr + step < n ? rank[curr + step] : -1));
tempRank[curr] = sameRank ? tempRank[prev] : tempRank[prev] + 1;
}
System.arraycopy(tempRank, 0, rank, 0, n);
if (rank[suffixArray[n - 1]] == n - 1) {
break;
}
}
return Arrays.stream(suffixArray).mapToInt(Integer::intValue).toArray();
}
}

View File

@@ -0,0 +1,44 @@
package com.thealgorithms.strings;
import static org.junit.jupiter.api.Assertions.assertArrayEquals;
import org.junit.jupiter.api.Test;
class SuffixArrayTest {
@Test
void testEmptyString() {
int[] result = SuffixArray.buildSuffixArray("");
assertArrayEquals(new int[] {}, result, "Empty string should return empty suffix array");
}
@Test
void testSingleCharacter() {
int[] result = SuffixArray.buildSuffixArray("a");
assertArrayEquals(new int[] {0}, result, "Single char string should return [0]");
}
@Test
void testDistinctCharacters() {
int[] result = SuffixArray.buildSuffixArray("abc");
assertArrayEquals(new int[] {0, 1, 2}, result, "Suffixes already in order for distinct chars");
}
@Test
void testBananaExample() {
int[] result = SuffixArray.buildSuffixArray("banana");
assertArrayEquals(new int[] {5, 3, 1, 0, 4, 2}, result, "Suffix array of 'banana' should be [5,3,1,0,4,2]");
}
@Test
void testStringWithDuplicates() {
int[] result = SuffixArray.buildSuffixArray("aaaa");
assertArrayEquals(new int[] {3, 2, 1, 0}, result, "Suffix array should be descending indices for 'aaaa'");
}
@Test
void testRandomString() {
int[] result = SuffixArray.buildSuffixArray("mississippi");
assertArrayEquals(new int[] {10, 7, 4, 1, 0, 9, 8, 6, 3, 5, 2}, result, "Suffix array for 'mississippi' should match expected");
}
}