edu.princeton.cs.algs4

Class TrieSET

java.lang.Object

edu.princeton.cs.algs4.TrieSET

All Implemented Interfaces:

java.lang.Iterable<java.lang.String>

public class TrieSET
extends java.lang.Object
implements java.lang.Iterable<java.lang.String>

The TrieSET class represents an ordered set of strings over the extended ASCII alphabet. It supports the usual add, contains, and delete methods. It also provides character-based methods for finding the string in the set that is the longest prefix of a given prefix, finding all strings in the set that start with a given prefix, and finding all strings in the set that match a given pattern.

This implementation uses a 256-way trie. The add, contains, delete, and longest prefix methods take time proportional to the length of the key (in the worst case). Construction takes constant time.

For additional documentation, see Section 5.2 of Algorithms in Java, 4th Edition by Robert Sedgewick and Kevin Wayne.

Constructor Summary

Constructors

Constructor and Description
TrieSET() Initializes an empty set of strings.

Method Summary

Modifier and Type	Method and Description
void	add(java.lang.String key) Adds the key to the set if it is not already present.
boolean	contains(java.lang.String key) Does the set contain the given key?
void	delete(java.lang.String key) Removes the key from the set if the key is present.
boolean	isEmpty() Is the set empty?
java.util.Iterator<java.lang.String>	iterator() Returns all of the keys in the set, as an iterator.
java.lang.Iterable<java.lang.String>	keysThatMatch(java.lang.String pattern) Returns all of the keys in the set that match pattern, where .
java.lang.Iterable<java.lang.String>	keysWithPrefix(java.lang.String prefix) Returns all of the keys in the set that start with prefix.
java.lang.String	longestPrefixOf(java.lang.String query) Returns the string in the set that is the longest prefix of query, or null, if no such string.
static void	main(java.lang.String[] args) Unit tests the TrieSET data type.
int	size() Returns the number of strings in the set.

Methods inherited from class java.lang.Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Methods inherited from interface java.lang.Iterable

forEach, spliterator

Constructor Detail

TrieSET

public TrieSET()

Initializes an empty set of strings.

Method Detail

contains

public boolean contains(java.lang.String key)

Does the set contain the given key?

Parameters:

key - the key

Returns:

true if the set contains key and false otherwise

Throws:

java.lang.IllegalArgumentException - if key is null

add

public void add(java.lang.String key)

Adds the key to the set if it is not already present.

Parameters:

key - the key to add

Throws:

java.lang.IllegalArgumentException - if key is null

size

public int size()

Returns the number of strings in the set.

Returns:

the number of strings in the set

isEmpty

public boolean isEmpty()

Is the set empty?

Returns:

true if the set is empty, and false otherwise

iterator

public java.util.Iterator<java.lang.String> iterator()

Returns all of the keys in the set, as an iterator. To iterate over all of the keys in a set named set, use the foreach notation: for (Key key : set).

Specified by:

iterator in interface java.lang.Iterable<java.lang.String>

Returns:

an iterator to all of the keys in the set

keysWithPrefix

public java.lang.Iterable<java.lang.String> keysWithPrefix(java.lang.String prefix)

Returns all of the keys in the set that start with prefix.

Parameters:

prefix - the prefix

Returns:

all of the keys in the set that start with prefix, as an iterable

keysThatMatch

public java.lang.Iterable<java.lang.String> keysThatMatch(java.lang.String pattern)

Returns all of the keys in the set that match pattern, where . symbol is treated as a wildcard character.

Parameters:

pattern - the pattern

Returns:

all of the keys in the set that match pattern, as an iterable, where . is treated as a wildcard character.

longestPrefixOf

public java.lang.String longestPrefixOf(java.lang.String query)

Returns the string in the set that is the longest prefix of query, or null, if no such string.

Parameters:

query - the query string

Returns:

the string in the set that is the longest prefix of query, or null if no such string

Throws:

java.lang.IllegalArgumentException - if query is null

delete

public void delete(java.lang.String key)

Removes the key from the set if the key is present.

Parameters:

key - the key

Throws:

java.lang.IllegalArgumentException - if key is null

main

public static void main(java.lang.String[] args)

Unit tests the TrieSET data type.

Parameters:

args - the command-line arguments