算法篇:前缀树

时间:2022-07-22
本文章向大家介绍算法篇:前缀树,主要内容包括其使用实例、应用技巧、基本知识点总结和需要注意事项,具有一定的参考价值,需要的朋友可以参考一下。

算法:

前缀树主要用于搜索算法,思想和实现并不复杂,属于典型题目,算法如下所示:

1.最多 n个指向子结点的链接,其中每个链接对应字母表数据集中的一个字母,
本文中假定 n 为 26,小写拉丁字母的数量。
2.布尔字段,以指定节点是对应键的结尾还是只是键前缀

题目:

https://leetcode-cn.com/problems/implement-trie-prefix-tree/

代码实现:

type Trie struct {
    IsEnd bool 
    Link [26]*Trie
}


/** Initialize your data structure here. */
func Constructor() Trie {
    root := Trie{}
    return  root
}

/** Inserts a word into the trie. */
func (this *Trie) Insert(word string)  {
    var node *Trie = this
    for _, w := range []byte(word) {
        if node.Link[w - byte('a')] == nil {
            n := Trie{}
            node.Link[w - byte('a')] = &n
        }
        node = node.Link[w - byte('a')]
    }
    node.IsEnd = true
    return
}


/** Returns if the word is in the trie. */
func (this *Trie) Search(word string) bool {
    var node *Trie = this 
    for _, w := range []byte(word) {
        if node.Link[w - byte('a')] == nil {
            node = nil
            break
        }
        node = node.Link[w - byte('a')]
    }
    return node != nil && node.IsEnd
}


/** Returns if there is any word in the trie that starts with the given prefix. */
func (this *Trie) StartsWith(prefix string) bool {
    var node *Trie = this 
    for _, w := range []byte(prefix) {
        if node.Link[w - byte('a')] == nil {
            node = nil
            break
        }
        node = node.Link[w - byte('a')]
    }
    return node != nil
}


/**
 * Your Trie object will be instantiated and called as such:
 * obj := Constructor();
 * obj.Insert(word);
 * param_2 := obj.Search(word);
 * param_3 := obj.StartsWith(prefix);
 */

执行结果: