Fixes bug where removal method was not working properly for cases where the word to be removed has no children for it's last node.

kelvinlauKL · kelvinlauKL · commit ed3a3218413b · 2016-12-06T15:29:19.000-08:00
diff --git a/Trie/Trie.playground/Sources/Trie.swift b/Trie/Trie.playground/Sources/Trie.swift
@@ -73,11 +73,8 @@ public extension Trie {
       currentNode.isTerminating = false
     } else {
       var character = currentNode.value
-      while currentNode.children.count == 0, let parent = currentNode.parent, !parent.isTerminating {
-        currentNode = parent
-        currentNode.children[character!] = nil
-        character = currentNode.value
-      }
+      currentNode.parent?.children[character!] = nil
+      character = currentNode.value
     }
   }
 }

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