关注互联网应用及运维技术的个人博客

Golang之Trie写法以及sort.Interface的使用

Trie

Trie又称字典树,是N叉树的一种特殊形式。是一种存储大量字符串的树形数据结构,相当于HashMap,在存储单词中又比HashMap节省了很多的内存空间。trie树的根节点不存储内容,内容都在孩子节点上。

这是一个保存了8个单词的字典树的结构,8个单词分别是”A”, “to”, “tea”, “ted”, “ten”, “i”, “in”, “inn”。每条从根节点到叶子节点的路径都构成了单词(有的不需要走到叶子节点也是单词,比如iin),从根节点出发的路径都是给定单词列表的某个/某些单词的前缀,反之如果在某个字符串中没有出现在这棵树的路径上,就说明组不成单词。

通常Trie树也会用在搜索引擎上,就是输入某些词,就会扩展出一些模糊词汇

Golang 实现的写法如下:

type Trie struct {
    children [26]*Trie
    isWord bool
}

// Insert
func (t *Trie) Insert(word string)  {
    cur := t
    for k, v := range word {
        c := v - 'a'
        if cur.children[c] == nil {
            cur.children[c] = &Trie{}
        }
        cur = cur.children[c]
        // end
        if k == len(word) -1 {
            cur.isWord = true
        }
    }
}

// Search
func (t *Trie) Search(word string) bool  {
    cur := t
    for _, v := range word {
        c := v - 'a'
        if cur.children[c] == nil {
            return false
        }
        cur = cur.children[c]
    }
    return cur.isWord
}

// StartWith
func (t *Trie) StartWith(prefix string) bool {
    cur := t
    for _, v := range prefix {
        c := v - 'a'
        if cur.children[c] == nil {
            return false
        }
        cur = cur.children[c]
    }
    return true
}

func main() {
    trie := new(Trie)

    trie.Insert("inn")
    fmt.Println(trie.Search("inn"))   // true
    fmt.Println(trie.Search("in"))    // false
    fmt.Println(trie.StartWith("in")) // true
    fmt.Println(trie.StartWith("ia")) // false
}

sort.Interface

关于自定义排序,用这个就可以了

在官方库sort.go中关于Interface定义如下:

// A type, typically a collection, that satisfies sort.Interface can be
// sorted by the routines in this package. The methods require that the
// elements of the collection be enumerated by an integer index.
type Interface interface {
    // Len is the number of elements in the collection.
    Len() int
    // Less reports whether the element with
    // index i should sort before the element with index j.
    Less(i, j int) bool
    // Swap swaps the elements with indexes i and j.
    Swap(i, j int)
}

所以我们只需要实现上面三个方法就可以了,如下是一个按照数组中字符串长度排序的一个例子

package main

import (
    "fmt"
    "sort"
)

type MySliceSort []string

func (m MySliceSort) Len() int {
    return len(m)
}

func (m MySliceSort) Less(i, j int) bool {
    return len(m[i]) > len(m[j])
}

func (m MySliceSort) Swap(i, j int) {
    m[i], m[j] = m[j], m[i]
}

func main() {
    sliceSort := MySliceSort{
        "这是五个字",
        "我要变六个字",
        "三个字",
    }
    sort.Sort(sliceSort)
    fmt.Println(sliceSort) // [我要变六个字 这是五个字 三个字]
}
赞(0)
未经允许不得转载:飞天狒狒 » Golang之Trie写法以及sort.Interface的使用

评论 抢沙发

  • 昵称 (必填)
  • 邮箱 (必填)
  • 网址