python中的二进制搜索树不起作用
-
28-09-2019 - |
题
class Node:
'''represents a new node in the BST'''
def __init__(self,key):
self.key=key
self.disconnect()
def disconnect(self):
self.left=None;
self.right=None;
self.parent=None;
def __str__(self):
return 'node with kay %s'%self.key
class BST:
def __init__(self):
self.root=None
def insert(self,t):
'''inserts a new element into the tree'''
self.find_place(self.root,t)
def find_place(self,node,key):
"""finds the right place of the element recursively"""
if node is None:
node=Node(key)
print node
else:
if node.key > key:
find_place(node.left,key)
else:
find_place(node.right,key)
def test():
'''function to test if the BST is working correctly'''
我编写了以上代码来实现二进制搜索树,但是插入方法无法正常工作,它甚至无法添加根元素。我无法解开原因。
解决方案
您实际上并不是在树上添加任何节点!
它最容易明确地管理根节点的添加,就像您在下面所做的那样 insert
.
一种 find_place
函数大概是从名称中返回父节点的功能,以及键是钥匙的左或右插槽吗?我已经明确了 _do_insert
在这两个步行和执行插入物的下方功能。
从那时起,您需要每次漫步在树上,或者您是否到达了一个空插槽,添加新节点。
重构代码可能是自然而然的 Node
班级。
在下面的代码中,我忽略了添加已经在树上的密钥,我只是默默退出:
def insert(self,t):
'''inserts a new element into the tree'''
if self.root is None:
self.root = Node(t)
else:
self._do_insert(self.root,t)
def _do_insert(self,parent,t):
if t > parent.key:
if parent.left is None:
parent.left = Node(t)
else:
self._do_insert(parent.left,t)
elif t < parent.key:
if parent.right is None:
parent.right = Node(t)
else:
self._do_insert(parent.right,t)
else:
# raise a KeyError or something appropriate?
pass
其他提示
这是python的另一个BST,使用分类键
左= 0右= 1值= 2 sort_key = -1
类BinarySearchtree(对象):
def __init__(self, sort_key=None):
self._root = []
self._sort_key = sort_key
self._len = 0
def insert(self,val):如果self._sort_key是无:sort_key = val //如果没有排序键,则sort key是value else:sort_key = self.__sort_key(val)
node = self._root
while node:
if sort_key < node[_SORT_KEY]:
node = node[LEFT]
else:
node = node[RIGHT]
if sort_key is val:
node[:] = [[], [], val]
else:
node[:] = [[], [], val, sort_key]
self._len += 1
def minimum(self):
return self._extreme_node(LEFT)[VALUE]
def maximum(self):
return self._extreme_node(RIGHT)[VALUE]
def find(self, sort_key):
return self._find(sort_key)[VALUE]
def _extreme_node(self, side):
if not self._root:
raise IndexError('Empty')
node = self._root
while node[side]:
node = node[side]
return node
def _find(self, sort_key):
node = self._root
while node:
node_key = node[SORT_KEY]
if sort_key < node_key:
node = node[LEFT]
elif sort_key > node_key:
node = node[RIGHT]
else:
return node
raise KeyError("%r not found" % sort_key)
如果有的话,排序键被值替换。
不隶属于 StackOverflow