tire_tree 模糊匹配完成

Author: hero

tree/tire_tree.go

@@ -1,9 +1,8 @@
 package tree
 
-import "fmt"
-
 type Trie struct {
 	node   map[rune]*Trie
+	val    string
 	isLast bool
 }
 
@@ -19,6 +18,7 @@ func (t *Trie) Insert(word string) {
 		}
 		t = t.node[v] //找到节点，跳到下一个节点
 	}
+	t.val = word
 	t.isLast = true
 }
 
@@ -42,40 +42,29 @@ func (t *Trie) StartsWith(prefix string) bool {
 	return true
 }
 
-func (t *Trie) GetAll(prefix string) *[]interface{} {
-	data := new([]interface{})
-	node := map[rune]*Trie{}
-	for _, v := range prefix { //找到全匹配下的节点
-		if t.node[v] != nil {
-			if t.node[v].isLast { //判断当前节点是不是一个完整单词
-				*data = append(*data, prefix)
-			}
-			t = t.node[v] //跳到下一个节点
-			node = t.node
-
+//查找数据
+func (t *Trie) SearchNode(key string) (res []interface{}) {
+	root := t
+	for _, v := range key {
+		if v, ok := root.node[v]; ok {
+			root.node = v.node
+		} else {
+			return
 		}
 	}
-	getNodeAll(prefix, node, data)
-	return data
-}
-
-var a string
-
-func getNodeAll(prefix string, node map[rune]*Trie, data *[]interface{}) {
-	fmt.Println("---------------------- prefix ", prefix)
-	for k, v := range node {
-		fmt.Println("---------------------- prefix  v1 ", prefix)
-
-		prefix = prefix + string(k)
-		fmt.Println("---------------------- prefix v2 ", prefix)
-		//fmt.Println(string(k),prefix)
-		if v.isLast {
-			fmt.Println("---------------------- 获得结果", prefix)
-			*data = append(*data, prefix)
-		}
-		if v.node != nil {
-			fmt.Println("---------------------- getNodeAll", prefix)
-			getNodeAll(prefix, v.node, data)
+	var queue []*Trie
+	queue = append(queue, root)
+	for len(queue) > 0 {
+		var childQueue []*Trie
+		for _, v := range queue { //遍历这层的数据
+			if v.isLast {
+				res = append(res, v.val)
+			}
+			for _, vi := range v.node { //把下一次的数据进行遍历
+				childQueue = append(childQueue, vi)
+			}
 		}
+		queue = childQueue
 	}
+	return
 }

tree/tire_tree_test.go

@@ -4,10 +4,12 @@ import "testing"
 
 func TestTrie_Insert(t *testing.T) {
 	trie := Constructor()
-	trie.Insert("apple")
+	trie.Insert("abp")
 	t.Log(trie.Search("apple"))
 	t.Log(trie.Search("app"))
 	t.Log(trie.StartsWith("app"))
 	trie.Insert("abe")
-	t.Log(trie.GetAll("a"))
+	trie.Insert("aef")
+	t.Log(trie.SearchNode("ae"))
+
 }