搜索Trie实现

时间:2014-05-12 18:41:57

标签: java generics trie

我编写了一个通过Trie.java接口实现的代码StringTrie.java:

package il.ac.tau.cs.sw1.trie;
import java.util.Set;

/**
 * Represents a generic trie data structure, where K is the type of keys that
 * are saved in the trie, and V is the type of values that can be saved for each
 * key.
 */
public interface Trie<K, V> {

    /**
     * Adds the input key to the data structure with the given value. If the key
     * is invalid for this trie (implementation-dependent) an exception is
     * thrown.
     */
    void addKey(K key, V value);

    /**
     * Searches all the keys in the trie structure with the given prefix. If
     * there are no such keys - an empty Set is returned. Otherwise, all the values saved
     * for these keys are added to the output value set. If the prefix is invalid for this
     * trie (implementation-dependent) an exception is thrown.
     * 
     * @post the return value is not null
     */
    Set<V> searchByPrefix(K prefix);

}

以及通过TrieNode.java实现的StringTrieNode.java:

package il.ac.tau.cs.sw1.trie;

import java.util.List;
import java.util.Set;

/**
 * Represents a node in a generic trie, where K is the type of keys that are
 * saved in the trie, and V is the type of values that can be saved for each
 * key.
 */
public interface TrieNode<K, V> {

    /**
     * Adds the key suffix to this trie node, with the given value, by
     * recursively adding the suffix of this suffix to the relevant child node.
     * 
     * @pre suffix is a valid suffix for this trie (implementation-dependent)
     * @post if the key ends in this trie node, the input value is added to the
     *       set of values returned by getValues().
     */
    void addSuffix(K suffix, V value);

    /**
     * Returns the set of values for keys that end in this node
     */
    Set<V> getValues();

    /**
     * Returns the child trie node such that the input suffix is saved in it or
     * in its descendant
     * 
     * @pre suffix is a valid suffix for this trie (implementation-dependent)
     * @post if no key with such a suffix exists in the trie, or if suffix ends
     *       in the current node, return null; otherwise return the relevant
     *       child node
     */
    TrieNode<K, V> getNext(K suffix);

    /**
     * Returns a list with all the child trie nodes of this node. The list may
     * contain null values.
     * 
     * @post $ret != null
     */
    List<TrieNode<K, V>> getNexts();

}

现在,我已经实现了所有功能(正确,我希望),现在我只有Set searchByPrefix(K前缀)的问题;,我找不到有效的算法+我有构造函数的问题。用户称之为:Trie names = new StringTrie&lt;&gt;();所以第一个必须在那里。但只有第二个首字母的字段。可以吗?

1 个答案:

答案 0 :(得分:1)

我建议你使用现有的实现。

以下是我已多次使用过的非常好的示例:

https://github.com/rkapsi/patricia-trie

以下是使用前缀搜索的示例:

PatriciaTrie<String, String> trie = new PatriciaTrie<String, String>(StringKeyAnalyzer.CHAR);
trie.put("prefix1", value);
trie.put("prefix2", value);

map = trie.prefixMap("prefix");