2017-12-11 13 views
0
class Node: 
    def __init__(self,value = None): 
     self.value = value 
     self.left = None 
     self.right = None 

class BinarySearchTree: 
    def __init__(self): 
     self.root = None 

    def insert(self,value): 
     if self.root != None: 
      self._insert(self.root,value) 
     else: 
      self.root = Node(value) 

    def _insert(self,current,value): 
     if value < current.value: 
      if current.left == None: 
       current.left = Node(value) 
      else: 
       self._insert(current.left,value) 

     elif value > current.value: 
      if current.right == None: 
       current.right = Node(value) 
      else: 
       self._insert(current.right,value) 
     else: 
      print("Value already exist in tree.") 


    def delete(self,current,value): 
     if current == None: return current 
     elif value < current.value: current.left = self.delete(current.left,value) 
     elif value > current.value: current.right = self.delete(current.right,value) 
     else: 
      if current.left == None and current.right == None: 
       current = None # case 0 
      elif current.left == None: return current.right # case 1r 
      elif current.right == None: return current.left # case 1l 
      else: # case 2 
       temp = self.min(current.right) 
       current.value = temp 
       self.delete(current.right,temp) 
     return current 

    def min(self,current): 
     if current != None: 
      return self._min(current) 
     else: 
      return None 

    def _min(self,current): 
     if current.left != None: 
      return self._min(current.left) 
     else: 
      return current.value 

if __name__ == '__main__': 
    for i in (12,5,18,3,4,6): 
     tree.insert(i) 
    tree.delete(tree.root,12) 

こんにちは、私のバイナリ検索ツリーの削除機能は、2人の子供がいるノードでは機能しません。バイナリ検索ツリーの削除機能がPythonで動作しない

この問題は、削除機能の6行目と12行目で発生します。この場合、self.delete(current.right,temp)という行はcurrent = None # case 0になります。線current = None # case 0は現在の値を0に変更しません。これは非常に混乱しています。どのようなアイデアが間違っている?

+0

これは、関数に渡した値ではなく、ローカルの 'current'をNoneに設定するだけです。それはあなたが期待していたものだと思いますか? – Jeronimo

答えて

1

とdeleteメソッドにライン12を置き換えます

current.right = self.delete(current.right, temp) 

理由は単純です:現在=なしのみローカルな値に影響を与えていない、それはノード12の左の子か右の子に影響を与えることができません。