/*
 *  @Author : huangzj
 *  @Time : 2020/12/14 16:15
 *  @Description：
 */

package trie

import (
	"fmt"
	"testing"
)

func TestTrie(t *testing.T) {
	commonTest()         //常规功能测试
	sensitiveWordsTest() //敏感词汇屏蔽测试
	searchTrie()         //查找前缀匹配的所有字符串
	searchTrieVague()    //模糊查询前缀匹配的所有字符串
}

func searchTrieVague() {
	fmt.Println()
	fmt.Println("模糊查询 查找前缀匹配的所有字符串")
	trie := New()
	trie.insert("abcuuid").insert("abcuued").insert("abcMxns").insert("abssde").insert("avddfa")
	trie.insert("我爱中国").insert("我爱祖国").insert("我爱学习").insert("我是谁").insert("我在哪里")
	trie.insert("我是你").insert("我").insert("我喜欢学习").insert("我在厦门").insert("我是一个程序员")

	fmt.Println()
	fmt.Println()
	result3, _ := trie.SearchTrieVague("我爱她")
	for _, r := range result3 {
		fmt.Println(r)
	}
}

func searchTrie() {
	fmt.Println()
	fmt.Println("查找前缀匹配的所有字符串")
	trie := New()
	trie.insert("abcuuid").insert("abcuued").insert("abcMxns").insert("abssde").insert("avddfa")
	trie.insert("我爱中国").insert("我爱祖国").insert("我爱学习").insert("我是谁").insert("我在哪里")

	fmt.Println()
	fmt.Println()
	result, _ := trie.SearchTrie("abc")
	for _, r := range result {
		fmt.Println(r)
	}

	fmt.Println()
	fmt.Println()
	result1, _ := trie.SearchTrie("我爱")
	for _, r := range result1 {
		fmt.Println(r)
	}

	fmt.Println()
	fmt.Println()
	result2, _ := trie.SearchTrie("我爱她")
	for _, r := range result2 {
		fmt.Println(r)
	}

}

func sensitiveWordsTest() {
	trie := New()
	trie.insert("粗话").insert("fuck").insert("fuckk").insert("脏话")
	s := "我是脏话，我是粗话，fuckk,fuck,fffuccck"
	word := trie.DealFunc(s, func(sentence []rune, start int, end int) string {
		var replace string
		for i := 0; i < end-start; i++ {
			replace = replace + "*"
		}
		runes := append(sentence[0:start], []rune(replace)...)
		runes = append(runes, sentence[end:]...)
		return string(runes)
	})
	fmt.Println(word)
}

func commonTest() {
	trie := New()
	trie.insert("ABC").insert("AB").insert("ABE").insert("ABEX").insert("XYZ").insert("你").insert("你好").insert("你是谁")
	result := trie.Traverse()
	for _, r := range result {
		fmt.Println(r)
	}

	fmt.Println()
	fmt.Println("进行查找测试")
	fmt.Println(trie.Search("Abc"))
	fmt.Println(trie.Search("ABC"))
	fmt.Println(trie.Search("ABEX"))
	fmt.Println(trie.Search("ABEx"))
	fmt.Println(trie.Search("你好"))
	fmt.Println(trie.Search("你好吗"))

	fmt.Println("进行删除测试")
	trie.Delete("你").Delete("ABC")
	fmt.Println(trie.Search("你"))
	fmt.Println(trie.Search("你好"))
	fmt.Println(trie.Search("你是谁"))
	fmt.Println(trie.Search("AB"))
	fmt.Println(trie.Search("ABE"))
	fmt.Println(trie.Search("ABC"))

	fmt.Println()
	fmt.Println("再次遍历所有的字典")
	result = trie.Traverse()
	for _, r := range result {
		fmt.Println(r)
	}
}