2022.12.5前缀树

1 力扣题208:前缀树

1.1 题目描述

Trie（发音类似 “try”）或者说 前缀树
是一种树形数据结构，用于高效地存储和检索字符串数据集中的键。这一数据结构有相当多的应用情景，例如自动补完和拼写检查。

1.2 思路分析

前缀树的特点是前缀相同的字符串其树的前部分节点是相同的，构造前缀树的关键在于一个节点表示一个字母，而子节点一共有26种可能性，用长度为26的数组表示即可。一种可能性为null时表示无后继节点，一种可能性为node时还需要加个节点标记，若标记为1表明是终结字符，可构成一个字符串，否则表明是中间字符，可作为前缀但不可构成字符串。关键在于26长度的数组和节点终结符标记

1.3 代码实现

private Trie[] children;
    private boolean isEnd;

    public Trie() {
        children = new Trie[26];
        isEnd = false;
    }
    
    public void insert(String word) {
        Trie node = this;
        for (int i = 0; i < word.length(); i++) {
            char ch = word.charAt(i);
            int index = ch - 'a';
            if (node.children[index] == null) {
                node.children[index] = new Trie();
            }
            node = node.children[index];
        }
        node.isEnd = true;
    }
    
    public boolean search(String word) {
        Trie node = searchPrefix(word);
        return node != null && node.isEnd;
    }
    
    public boolean startsWith(String prefix) {
        return searchPrefix(prefix) != null;
    }

    private Trie searchPrefix(String prefix) {
        Trie node = this;
        for (int i = 0; i < prefix.length(); i++) {
            char ch = prefix.charAt(i);
            int index = ch - 'a';
            if (node.children[index] == null) {
                return null;
            }
            node = node.children[index];
        }
        return node;
    }