内存有问题吗？你可以转向久经考验的真实......TRIE！
构建后缀树！
鉴于您的意见['abc','abcd','ab','def','efgd']
我们会有一棵树
_
/ | \
a e d
/ | \
b* f e
/ | \
c* g f*
/ |
d* d*
利用所述树的 DFS（深度优先搜索）搜索，您将找到最深的叶子 abcd、efgd 和 def
树遍历非常简单，您的时间复杂度是 O(n*m). 比之前的 O(n^2) 时间有更好的改进。
使用这种方法，添加新键变得简单，并且仍然可以轻松找到唯一键。
考虑添加密钥deg
你的新树大约是
_
/ | \
a e d
/ | \
b* f e
/ | / \
c* g g* f*
/ |
d* d*
使用这个新树，执行 DFS 搜索以获取不是其他人前缀的唯一键仍然是一件简单的事情。
from typing import List
class Trie(object):
class Leaf(object):
def __init__(self, data, is_key):
self.data = data
self.is_key = is_key
self.children = []
def __str__(self):
return "{}{}".format(self.data, "*" if self.is_key else "")
def __init__(self, keys):
self.root = Trie.Leaf('', False)
for key in keys:
self.add_key(key)
def add_key(self, key):
self._add(key, self.root.children)
def has_suffix(self, suffix):
leaf = self._find(suffix, self.root.children)
if not leaf:
return False
# This is only a suffix if the returned leaf has children and itself is not a key
if not leaf.is_key and leaf.children:
return True
return False
def includes_key(self, key):
leaf = self._find(key, self.root.children)
if not leaf:
return False
return leaf.is_key
def delete(self, key):
"""
If the key is present as a unique key as in it does not have any children nor are any of its nodes comprised of
we should delete all of the nodes up to the root
If the key is a prefix of another long key in the trie, umark the leaf node
if the key is present in the trie and contains no children but contains nodes that are keys we should delete all
nodes up to the first encountered key
:param key:
:return:
"""
if not key:
raise KeyError
self._delete(key, self.root.children, None)
def _delete(self, key, children: List[Leaf], parents: (List[Leaf], None), key_idx=0, parent_key=False):
if not parents:
parents = [self.root]
if key_idx >= len(key):
return
key_end = True if len(key) == key_idx + 1 else False
suffix = key[key_idx]
for leaf in children:
if leaf.data == suffix:
# we have encountered a leaf node that is a key we can't delete these
# this means our key shares a common branch
if leaf.is_key:
parent_key = True
if key_end and leaf.children:
# We've encountered another key along the way
if parent_key:
leaf.is_key = False
else:
# delete all nodes recursively up to the top of the first node that has multiple children
self._clean_parents(key, key_idx, parents)
elif key_end and not leaf.children:
# delete all nodes recursively up to the top of the first node that has multiple children
self._clean_parents(key, key_idx, parents)
# Not at the key end so we need to keep traversing the tree down
parents.append(leaf)
self._delete(key, leaf.children, parents, key_idx + 1, key_end)
def _clean_parents(self, key, key_idx, parents):
stop = False
while parents and not stop:
p = parents.pop()
# Need to stop processing a removal at a branch
if len(p.children) > 1:
stop = True
# Locate our branch and kill its children
for i in range(len(p.children)):
if p.children[i].data == key[key_idx]:
p.children.pop(i)
break
key_idx -= 1
def _find(self, key, children: List[Leaf]):
if not key:
raise KeyError
match = False
if len(key) == 1:
match = True
suffix = key[0]
for leaf in children:
if leaf.data == suffix and not match:
return self._find(key[1:], leaf.children)
elif leaf.data == suffix and match:
return leaf
return None
def _add(self, key, children: List[Leaf]):
if not key:
return
is_key = False
if len(key) == 1:
is_key = True
suffix = key[0]
for leaf in children:
if leaf.data == suffix:
self._add(key[1:], leaf.children)
break
else:
children.append(Trie.Leaf(suffix, is_key))
self._add(key[1:], children[-1].children)
return
@staticmethod
def _has_children(leaf):
return bool(leaf.children)
def main():
keys = ['ba', 'bag', 'a', 'abc', 'abcd', 'abd', 'xyz']
trie = Trie(keys)
print(trie.includes_key('ba')) # True
print(trie.includes_key('b')) # False
print(trie.includes_key('dog')) # False
print(trie.has_suffix('b')) # True
print(trie.has_suffix('ab')) # True
print(trie.has_suffix('abd')) # False
trie.delete('abd') # Should only remove the d
trie.delete('a') # should unmark a as a key
trie.delete('ba') # should remove the ba trie
trie.delete('xyz') # Should remove the entire branch
trie.delete('bag') # should only remove the g
print(trie)
if __name__ == "__main__":
main()
请注意，上面的 trie 实现没有实现 DFS 搜索；但是，它为您提供了一些惊人的开始。