算法随笔-前缀树

208. 实现 Trie (前缀树)

实现上到不是很难，我们用数组下标代替hashmap，用于存放有限切固定长度的映射关系，用一个bool变量表示当前位置上是不是合法的word而不是前缀。

有一个很容易绕进去的地方是，如何表示树的root

我们希望引用一个地址，持久性的保持"Trie的树根"

其实在内部用this即可，不需要在Trie里弄一个Trie的对象，这会导致stackoverflow... 一时半会没绕过来弯

class Trie {

    // 自己引用自己, java是可以的，因为java本质上对象是pointer类型
    // c++的话必须显式声明为Trie*, 否则就是 incomplete class
    
    // 是否在当前位置存在一个单词
    public boolean isFin = false;
    // 全部初始化为null
    public Trie[] ts = new Trie[26];

    public Trie() {
    }
    
    public void insert(String word) {
        char[] cs = word.toCharArray();
        Trie cur = this;
        for (int i = 0; i < cs.length; i++) {
            int idx = cs[i] - 'a';
            if (cur.ts[idx] == null) {
                cur.ts[idx] = new Trie();
            }
            cur = cur.ts[idx];

            if (i == cs.length - 1) {
                // 最后一个的话，此时ts[idx]的isFin设置为true
                cur.isFin = true;
            }
        }
    }
    
    public boolean search(String word) {
        char[] cs = word.toCharArray();

        Trie cur = this;
        for (char c : cs) {
            int idx = c - 'a';
            if (cur.ts[idx] == null) return false;
            cur = cur.ts[idx];
        }
        // 当前单词是一个单词的前缀
        if (cur.isFin == false) return false; 

        return true;
    }
    
    public boolean startsWith(String prefix) {
        char[] cs = prefix.toCharArray();

        Trie cur = this;
        for (char c : cs) {
            int idx = c - 'a';
            if (cur.ts[idx] == null) return false;
            cur = cur.ts[idx];
        }
        // 当前单词是一个单词的前缀
        return true;
    }
}

/**
 * Your Trie object will be instantiated and called as such:
 * Trie obj = new Trie();
 * obj.insert(word);
 * boolean param_2 = obj.search(word);
 * boolean param_3 = obj.startsWith(prefix);
 */