Rotazione destra e sinistra di un albero in pitone
Uso la classe:
class Node:
def __init__(self, value):
self.key = value
self.left = None
self.right = None
self.parent = None
e ho creato questo albero:
n_12 = Node(12)
n_15 = Node(15)
n_3 = Node(3)
n_7 = Node(7)
n_1 = Node(1)
n_2 = Node(2)
n_not1 = Node(-1)
n_12.right = n_15
n_12.left = n_3
n_3.right = n_7
n_3.left = n_1
n_1.right = n_2
n_1.left = n_not1
n_12.parent = None
n_15.parent = n_12
n_3.parent = n_12
n_7.parent = n_3
n_1.parent = n_3
n_2.parent = n_1
n_not1.parent = n_1
Ho provato questo codice:
def rightRotate(t):
if t == None or t.left == None:
return None
n = t
l = t.left
r = t.right
lr = t.left.right
ll = t.left.left
t = t.left
t.right = n
if r != None:
t.right.right = r
if lr != None:
t.right.left = lr
if ll != None:
t.left = ll
Ma non ha funzionato, utilizzando il nodo radice n_12
cancella alcuni nodi. Perché non ha funzionato e non capisco perché non ho tutti i nodi. Se chiamo rightRotate(n_1)
, ho un ciclo infinito.
Risposte
Scrivi "Ho un loop infinito" , ma il tuo codice non ha loop, quindi deve accadere altrove nel tuo codice.
Vedo due problemi:
1) L'assegnazione dovrebbe essere incondizionata
if lr != None:
t.right.left = lr
Questo incarico è necessario anche quando lr is None
. In caso contrario, t.right.left
rimarrà uguale a quello l
che è t
in quel momento, e così rimarrai davvero con un loop nel tuo albero.
2) Doppia filettatura
Il tuo albero è a doppio thread, cioè ha anche parent
collegamenti. Ma questi non sono aggiornati nella tua rightRotate
funzione. Quindi o fare a meno dei parent
collegamenti (che è preferibile) o adattare il codice in modo che anche i parent
collegamenti vengano aggiornati in base alla rotazione.
Altre osservazioni:
Il seguente pezzo di codice potrebbe essere semplificato:
if r != None:
t.right.right = r # was already equal to r
if lr != None:
t.right.left = lr # see above. should not be behind a condition
if ll != None:
t.left = ll # was already equal to ll
In modo che possa essere ridotto a solo:
t.right.left = lr
o anche:
n.left = lr
Codice finale
Con le modifiche sopra menzionate, la tua funzione potrebbe essere:
class Node:
def __init__(self, value):
self.key = value
self.left = None
self.right = None
self.parent = None
def rightRotate(node):
if node is None or node.left is None:
return node
parent = node.parent
left = node.left
left_right = left.right
# change edge 1
if parent: # find out if node is a left or right child of node
if parent.left == node:
parent.left = left
else:
parent.right = left
left.parent = parent
# change edge 2
left.right = node
node.parent = left
# change edge 3
node.left = left_right
if left_right:
left_right.parent = node
return left # the node that took the position of node
# your code to build the tree
n_12 = Node(12)
n_15 = Node(15)
n_3 = Node(3)
n_7 = Node(7)
n_1 = Node(1)
n_2 = Node(2)
n_not1 = Node(-1)
n_12.right = n_15
n_12.left = n_3
n_3.right = n_7
n_3.left = n_1
n_1.right = n_2
n_1.left = n_not1
n_12.parent = None
n_15.parent = n_12
n_3.parent = n_12
n_7.parent = n_3
n_1.parent = n_3
n_2.parent = n_1
n_not1.parent = n_1
# rotate the root
root = n_12
root = rightRotate(root) # returns the node that took the place of n_12
Basta rimuovere le linee con parent
per ottenere la versione a thread singolo.