Oldingi darsda ko'rgan g'oyani endi Go kodiga aylantiramiz. Har bir tugun — bolalar xaritasi (map[rune]*TrieNode, chunki har bir "bola" muayyan bir harfga mos keladi) va isEnd bayrog'idan iborat bo'ladi.

package main

import "fmt"

type TrieNode struct {
	children map[rune]*TrieNode
	isEnd    bool
}

func newTrieNode() *TrieNode {
	return &TrieNode{children: make(map[rune]*TrieNode)}
}

type Trie struct {
	root *TrieNode
}

func NewTrie() *Trie {
	return &Trie{root: newTrieNode()}
}

func (t *Trie) Insert(word string) {
	node := t.root
	for _, ch := range word {
		if node.children[ch] == nil {
			node.children[ch] = newTrieNode()
		}
		node = node.children[ch]
	}
	node.isEnd = true
}

func main() {
	t := NewTrie()
	t.Insert("go")
	t.Insert("gopher")
	fmt.Println(t.root.children['g'].children['o'].isEnd)
}

Insert so'zning har bir harfi (rune) bo'yicha yuradi: agar joriy tugunda shu harf uchun "bola" hali yo'q bo'lsa, yangisini yaratadi; bor bo'lsa, borini ishlatadi. Sikl tugagach (barcha harflar "bosib o'tilgach"), oxirgi tugunga isEnd = true belgilanadi — "aynan shu yergacha bo'lgan yo'l to'liq bir so'z" deb belgilash.

"go" va "gopher" qo'shilgach, g -> o yo'li ikkalasida ham bir xil, umumiy tugunlardan iborat — Trie ularni ikki marta emas, bir marta saqlaydi. Shuning uchun root.children['g'].children['o'].isEnd — true, chunki "go" aynan shu yerda tugaydi (garchi undan keyin ham "gopher" uchun yo'l davom etsa ham).