diff --git a/data_structures/trie/radix_tree.py b/data_structures/trie/radix_tree.py index 11e0dc23faa9..be212971907d 100644 --- a/data_structures/trie/radix_tree.py +++ b/data_structures/trie/radix_tree.py @@ -118,6 +118,9 @@ def find(self, word: str) -> bool: >>> RadixNode("myprefix").find("mystring") False """ + if word == "": + return self.is_leaf + incoming_node = self.nodes.get(word[0], None) if not incoming_node: return False @@ -147,6 +150,11 @@ def delete(self, word: str) -> bool: >>> RadixNode("myprefix").delete("mystring") False """ + if word == "": + was_present = self.is_leaf + self.is_leaf = False + return was_present + incoming_node = self.nodes.get(word[0], None) if not incoming_node: return False