From 22d935d07d3cb2576ff9de84dacd7bf6256633d6 Mon Sep 17 00:00:00 2001
From: ziyitony <ziyitonyu@gmail.com>
Date: Thu, 16 Apr 2020 20:22:18 +0900
Subject: [PATCH] update add() function to make use of range keyword

---
 trie_tree.go | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/trie_tree.go b/trie_tree.go
index d17ea66..600a436 100644
--- a/trie_tree.go
+++ b/trie_tree.go
@@ -30,8 +30,7 @@ func (tree *Trie) Add(words ...string) {
 func (tree *Trie) add(word string) {
 	var current = tree.Root
 	var runes = []rune(word)
-	for position := 0; position < len(runes); position++ {
-		r := runes[position]
+	for _, r := range runes {
 		if next, ok := current.Children[r]; ok {
 			current = next
 		} else {
@@ -39,10 +38,9 @@ func (tree *Trie) add(word string) {
 			current.Children[r] = newNode
 			current = newNode
 		}
-		if position == len(runes)-1 {
-			current.isPathEnd = true
-		}
 	}
+
+	current.isPathEnd = true
 }
 
 func (tree *Trie) Del(words ...string) {