2 
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メソッドが期待どおりに機能せず、ルート要素も追加できません。原因がわからない。

4

2 に答える 2

2

実際にツリーにノードを追加しているわけではありません。

以下ので行ったように、ルートノードの追加を明示的に管理するのが最も簡単です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
于 2010-06-20T07:51:05.590 に答える
0

これは、ソートキーを使用したPythonの別のBSTです

LEFT = 0 RIGHT = 1 VALUE = 2 SORT_KEY = -1

クラスBinarySearchTree（object）：

def __init__(self, sort_key=None):
    self._root = []  
    self._sort_key = sort_key
    self._len = 0

def insert（self、val）：self._sort_keyがNoneの場合：sort_key = val //ソートキーがない場合、sort keyは値です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)

ソートキーは、存在する場合は値に置き換えられます。

于 2013-04-29T12:25:55.547 に答える