Fixed a bug where remove method only clears the last node.

kelvinlauKL · kelvinlauKL · commit dcf9d22a82fb · 2016-12-07T16:42:54.000-08:00
diff --git a/Trie/Trie.playground/Sources/Trie.swift b/Trie/Trie.playground/Sources/Trie.swift
@@ -73,8 +73,16 @@ public extension Trie {
       currentNode.isTerminating = false
     } else {
       var character = currentNode.value
-      currentNode.parent?.children[character!] = nil
-      character = currentNode.value
+      
+      while currentNode.children.count == 0, let parent = currentNode.parent {
+        currentNode = parent
+        currentNode.children[character!] = nil
+        character = currentNode.value
+        
+        if currentNode.isTerminating {
+          break
+        }
+      }
     }
   }
 }

Original file line number	Diff line number	Diff line change
`@@ -73,8 +73,16 @@ public extension Trie {`
`73`	`73`	`currentNode.isTerminating = false`
`74`	`74`	`} else {`
`75`	`75`	`var character = currentNode.value`
`76`		`- currentNode.parent?.children[character!] = nil`
`77`		`- character = currentNode.value`
	`76`	`+`
	`77`	`+ while currentNode.children.count == 0, let parent = currentNode.parent {`
	`78`	`+ currentNode = parent`
	`79`	`+ currentNode.children[character!] = nil`
	`80`	`+ character = currentNode.value`
	`81`	`+`
	`82`	`+ if currentNode.isTerminating {`
	`83`	`+ break`
	`84`	`+ }`
	`85`	`+ }`
`78`	`86`	`}`
`79`	`87`	`}`
`80`	`88`	`}`