Binar Arborele de căutare în Python nu funcționează

Question

Binar Arborele de căutare în Python nu funcționează

voturi

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'''

Am scris codul de mai sus pentru a pune în aplicare un arbore binar de căutare, dar metoda de inserție nu funcționează conform așteptărilor, nu reușește să adăugați chiar elementul rădăcină. Nu se poate undestand cauza.

python algorithm binary-search-tree

Întrebat 20/06/2010 la 08:40
sursa de către utilizator Bunny Rabbit

În alte limbi...

2 răspunsuri

voturi

0

Aici este un alt BST cu Python, folosind o cheie de sortare

STÂNGA = 0 DREAPTA = 1 VALOARE = 2 SORT_KEY = -1

class BinarySearchTree (obiect):

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

inserați def (de sine, val): în cazul în care este self._sort_key Fără: sort_key = val // în cazul în care nici o cheie de sortare, sortare cheie este valoarea altceva: 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)

Tasta de sortare este înlocuită cu o valoare dacă este cazul.

Publicat 29/04/2013 la 13:25
sursa de către utilizator octoback

Will · Accepted Answer · 2010-06-20 08:51

Nu esti de fapt, adăugarea de orice noduri la copac!

Cel mai simplu de a gestiona adăugarea a nodului rădăcină în mod explicit, cum vedeți că am făcut mai jos , în insert.

O find_placefuncție ar, probabil de la numele, întoarce nodul părinte și , de asemenea , dacă este stânga sau slotul din dreapta pentru cheia? Am făcut o explicită _do_insertfuncție de mai jos că atât plimbări și Insertul.

De atunci, trebuie să meargă copac, de fiecare dată când văd dacă te recursiv în jos o sucursală sau dacă ați atins un slot gol, în cazul în care adăugați noul nod.

S - ar putea fi firesc să refactor codul pentru a pune responsabilitatea pentru mersul pe jos copac (și de a face adaugă, elimină și astfel) în Nodeclasa.

În codul de mai jos, am ignora adăugând o cheie care se află deja în copac, am doar tăcere ieși din:

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