前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >算法篇:前缀树

算法篇:前缀树

作者头像
灰子学技术
发布2020-08-06 11:49:51
4000
发布2020-08-06 11:49:51
举报
文章被收录于专栏:灰子学技术灰子学技术

算法:

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

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

题目:

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

代码实现:

代码语言:javascript
复制
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);
 */

执行结果:

本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2020-08-03,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 灰子学技术 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档