数据结构之前缀树

前缀树

前缀树 是 N叉树 的一种特殊形式。通常来说，一个前缀树是用来存储字符串的。前缀树的每一个节点代表一个字符串（前缀）。每一个节点会有多个子节点，通往不同子节点的路径上有着不同的字符。子节点代表的字符串是由节点本身的原始字符串 ，以及通往该子节点路径上所有的字符组成的。下面是前缀树的一个例子：

树节点：

// 节点类
static class TrieNode {
    // 单词标记
    private boolean isWord;
    // 字符表
    private Map<Character, TrieNode> childrenMap = new HashMap<>();
}

插入前缀

当我们在二叉搜索树中插入目标值时，在每个节点中，我们都需要根据节点值和目标值之间的关系，来确定目标值需要去往哪个子节点。同样地，当我们向前缀树中插入一个目标值时，我们也需要根据插入的目标值来决定我们的路径。

更具体地说，如果我们在前缀树中插入一个字符串 S，我们要从根节点开始。 我们将根据 S[0]（S中的第一个字符），选择一个子节点或添加一个新的子节点。然后到达第二个节点，并根据 S[1] 做出选择。 再到第三个节点，以此类推。 最后，我们依次遍历 S 中的所有字符并到达末尾。 末端节点将是表示字符串 S 的节点。例：

搜索前缀

正如我们在前缀树的简介中提到的，所有节点的后代都与该节点相对应字符串的有着共同前缀。因此，很容易搜索以特定前缀开头的任何单词。

同样地，我们可以根据给定的前缀沿着树形结构搜索下去。一旦我们找不到我们想要的子节点，搜索就以失败终止。否则，搜索成功。为了更具体地解释搜索的过程，我们提供了下列示例：

/**
* 如果之前已经插入的字符串 word 的前缀之一为 prefix ，返回 true ；否则，返回 false 。
*
* @param prefix 前缀
* @return
*/
public boolean startsWith(String prefix) {
    TrieNode cur = root;
    //若循环正常执行完，表示所有的前缀都存在，直接返回true
    for (int i = 0; i < prefix.length(); i++) {
        char c = prefix.charAt(i);
        //若查找的前缀不存在，则直接返回false
        if (cur.childrenMap.get(c) == null) {
            return false;
        }
        //移动到下一个节点对象
        cur = cur.childrenMap.get(c);
    }
    return true;
}

搜索单词

你可能还想知道如何搜索特定的单词，而不是前缀。我们可以将这个词作为前缀，并同样按照上述同样的方法进行搜索。如果搜索失败，那么意味着没有单词以目标单词开头，那么目标单词绝对不会存在于前缀树中。如果搜索成功，我们需要检查目标单词是否是前缀树中单词的前缀，或者它本身就是一个单词。为了进一步解决这个问题，你可能需要稍对节点的结构做出修改。

/**
* 查找指定的前缀是否存在
*
* @param word 字符串单词
* @return 若字符串work在前缀树中，返回true，否则返回false
*/
public boolean search(String word) {
    TrieNode cur = root;
    //若循环正常执行完，表示所有的前缀都存在，根据isWord属性值来判断是否为一个单词
    for (int i = 0; i < word.length(); i++) {
        char c = word.charAt(i);
        //若查找的前缀不存在，则直接返回false
        if (cur.childrenMap.get(c) == null) {
            return false;
        }
        //移动到下一个节点对象
        cur = cur.childrenMap.get(c);
    }
    return cur.isWord;
}

前缀树代码实现

/**
 * Author : AiTao
 * Date : 2020/11/9
 * Time : 19:56
 * Information : 前缀树
 */
public class PrefixTrie {
    // 节点类
    private static class TrieNode {
        // 单词标记
        private boolean isWord;
        // 字符表
        private Map<Character, TrieNode> childrenMap = new HashMap<>();
    }

    // 根节点
    private TrieNode root;

    //初始化前缀树对象
    public PrefixTrie() {
        root = new TrieNode();
    }

    /**
     * 向前缀树中插入字符串word
     *
     * @param word 字符串单词
     */
    public void insert(String word) {
        TrieNode cur = root;
        for (int i = 0; i < word.length(); i++) {
            // 将字符串拆分成一个个字符
            char c = word.charAt(i);
            // 如果路径不存在，直接插入一个节点对象
            if (cur.childrenMap.get(c) == null) {
                cur.childrenMap.put(c, new TrieNode());
            }
            //移动到下一个节点对象
            cur = cur.childrenMap.get(c);
        }
        cur.isWord = true;
    }

    /**
     * 查找指定的前缀是否存在
     *
     * @param word 字符串单词
     * @return 若字符串work在前缀树中，返回true，否则返回false
     */
    public boolean search(String word) {
        TrieNode cur = root;
        //若循环正常执行完，表示所有的前缀都存在，根据isWord属性值来判断是否为一个单词
        for (int i = 0; i < word.length(); i++) {
            char c = word.charAt(i);
            //若查找的前缀不存在，则直接返回false
            if (cur.childrenMap.get(c) == null) {
                return false;
            }
            //移动到下一个节点对象
            cur = cur.childrenMap.get(c);
        }
        return cur.isWord;
    }

    /**
     * 如果之前已经插入的字符串 word 的前缀之一为 prefix ，返回 true ；否则，返回 false 。
     *
     * @param prefix 前缀
     * @return
     */
    public boolean startsWith(String prefix) {
        TrieNode cur = root;
        //若循环正常执行完，表示所有的前缀都存在，直接返回true
        for (int i = 0; i < prefix.length(); i++) {
            char c = prefix.charAt(i);
            //若查找的前缀不存在，则直接返回false
            if (cur.childrenMap.get(c) == null) {
                return false;
            }
            //移动到下一个节点对象
            cur = cur.childrenMap.get(c);
        }
        return true;
    }