208. 实现 Trie (前缀树) （中等题）

题目描述：
实现一个 Trie (前缀树)，包含 insert, search, 和 startsWith 这三个操作。

示例:

Trie trie = new Trie();

trie.insert(“apple”);
trie.search(“apple”); // 返回 true
trie.search(“app”); // 返回 false
trie.startsWith(“app”); // 返回 true
trie.insert(“app”);
trie.search(“app”); // 返回 true

说明:

你可以假设所有的输入都是由小写字母 a-z 构成的。
保证所有输入均为非空字符串。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/implement-trie-prefix-tree
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
解法：

class TrieNode{
    public int path;  	// 记录路过该点的数量
    public int end;		// 记录以该点结尾的数量
    public TrieNode[] nexts;	//记录26个节点中是否插入过

    public TrieNode(){
        path = 0;
        end = 0;
        nexts = new TrieNode[26];
    }
}

class Trie {
    private TrieNode root;

    public Trie() {
        root = new TrieNode();
    }

    public void insert(String word) {
        if(word == null){
            return;
        }
        char[] ch = word.toCharArray();
        TrieNode node = root;
        for (int i = 0; i < ch.length; i++) {
            int index = ch[i] - 'a';
            if(node.nexts[index] == null){//插入时，如果不是之前出现的单词，则添加新结点
                node.nexts[index] = new TrieNode();
            }
            node = node.nexts[index];//之后结点后移
            node.path++;	//使当前结点的path增加1
        }
        node.end++;//使结尾结点的end增加1
    }

    public boolean search(String word) {
        if(word == null){
            return false;
        }
        char[] ch = word.toCharArray();
        TrieNode node = root;
        for (int i = 0; i < ch.length; i++) {
            int index = ch[i] - 'a';
            if(node.nexts[index] == null){
                return false;
            }
            node = node.nexts[index];
        }
        return node.end > 0 ;
    }

    public boolean startsWith(String prefix) {
        if(prefix == null){
            return false;
        }
        TrieNode node = root;
        char[] ch = prefix.toCharArray();
        for (int i = 0; i < ch.length; i++) {
            int index = ch[i] - 'a';
            if(node.nexts[index] == null){
                return false;
            }
            node = node.nexts[index];
        }
        return node.path > 0;
    }
}

思路概述：
构造TrieNode已经写在注释里了，查找是否存在与插入基本相同，只是在不断向下寻找的过程中，遇到空则直接返回false。遍历结束后，返回该结点的end值是否大于 1 。
同样的，查找是否以某单词开头，则是返回该结点的path值是否大于 1.
附带一个题目里未要求的删除操作

    public void delete(String word){
        if(search(word) != false){ //先检查该单词是否存在
            char[] ch = word.toCharArray();
            TrieNode node = root;
            for (int i = 0; i < ch.length; i++) {
                int index = ch[i] - 'a';
                if(--node.nexts[index].path == 0){//如果该单词之后的结点都只出现一次，
                    node.nexts[index] = null;	// 可以直接将后面全部舍去
                    return;
                }
                node = node.nexts[index];
            }
            node.end--;// 最后将末尾结点的end值减一
        }
    }