mathdatech/Python
1
0
Fork 0
mirror of https://github.com/TheAlgorithms/Python.git synced 2024-11-27 23:11:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

rotate left almost working

Browse Source
This commit is contained in:
Rafael 2017-07-08 01:26:49 +02:00
parent 8384cbb879
commit 67f23f59b1
1 changed files with 129 additions and 48 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

177
data_structures/AVL/AVL.py
View File

@ -23,18 +23,24 @@ class Node:
def setLeft(self, left): def setLeft(self, left):
self.left = left self.left = left
self.left.setParent(self)
self.left.setHeight(self.height + 1)
def getRight(self): def getRight(self):
return self.rigt return self.rigt
def setRight(self, right): def setRight(self, right):
self.rigt = right self.rigt = right
self.rigt.setParent(self)
self.rigt.setHeight(self.height + 1)
def getParent(self): def getParent(self):
return self.parent return self.parent
def setParent(self, parent): def setParent(self, parent):
self.parent = parent self.parent = parent
self.height = (self.parent.getHeight() + 1 if
(self.parent is not None) else 0)
def setHeight(self, height): def setHeight(self, height):
self.height = height self.height = height
@ -53,6 +59,7 @@ class AVL:
node = Node(value) node = Node(value)
if self.root is None: if self.root is None:
self.root = node self.root = node
self.root.setHeight(0)
self.size = 1 self.size = 1
else: else:
# Same as Binary Tree # Same as Binary Tree
@ -71,54 +78,59 @@ class AVL:
else: else:
if node.getLabel() < dad_node.getLabel(): if node.getLabel() < dad_node.getLabel():
dad_node.setLeft(node) dad_node.setLeft(node)
dad_node.setHeight(dad_node.getHeight() + 1)
else: else:
dad_node.setRight(node) dad_node.setRight(node)
dad_node.setHeight(dad_node.getHeight() + 1) self.rebalance(dad_node)
break break
def rebalance(self, node): def rebalance(self, node):
height_right = 0 height_right = 0
height_left = 0 height_left = 0
n = node
while n is not None:
height_right = 0
height_left = 0
if node.getRight() is not None: if node.getRight() is not None:
height_right = node.getRight().getHeight() height_right = node.getRight().getHeight()
if node.getLeft() is not None: if node.getLeft() is not None:
height_left = node.getLeft().getHeight() height_left = node.getLeft().getHeight()
if abs(height_left - height_right) > 1: if abs(height_left - height_right) > 1:
if height_left > height_right: if height_left > height_right:
right_child = node.getRight() left_child = node.getRight()
if (right_child.getLeft().getHeight() > if ():
right_child.getRight().getHeight()): self.rotate_left(n)
self.rotate_left(node) break
else:
self.double_rotate_right(n)
break
else: else:
self.double_rotate_right(node) right_child = node.getRight()
else: if right_child is not None:
left_child = node.getLeft() h_right = (right_child.getRight().getHeight()
if (left_child.getLeft().getHeight() > if (right_child.getRight() is not None) else 0)
left_child.getRight().getHeight()): h_left = (right_child.getLeft().getHeight()
self.double_rotate_left(node) if (right_child.getLeft() is not None) else 0)
else:
self.rotate_right(node) if (h_left > h_right):
self.double_rotate_left(n)
break
else:
self.rotate_right(n)
print(n.getLabel())
break
n = n.getParent()
def rotate_left(self, node): def rotate_left(self, node):
# TODO: is this pythonic enought? pass
aux = node.getLabel()
node = aux.getRight()
node.setHeight(node.getHeight() - 1)
node.setLeft(Node(aux))
node.getLeft().setHeight(node.getHeight() + 1)
node.getRight().setHeight(node.getRight().getHeight() - 1)
def rotate_right(self, node): def rotate_right(self, node):
aux = node.getLabel() n = Node(node.getLabel())
node = aux.getLeft() n.setRight(node.getRight())
node.setHeight(node.getHeight() - 1) n.setLeft(Node(node.getParent().getLabel()))
node.setRight(Node(aux)) node = n
node.getLeft().setHeight(node.getHeight() + 1)
node.getLeft().setHeight(node.getLeft().getHeight() - 1)
def double_rotate_left(self, node): def double_rotate_left(self, node):
self.rotate_right(node.getRight().getRight()) self.rotate_right(node.getRight().getRight())
@ -139,23 +151,92 @@ class AVL:
print(curr_node.getLabel(), end=" ") print(curr_node.getLabel(), end=" ")
self.preShow(curr_node.getRight()) self.preShow(curr_node.getRight())
def preorden(self, curr_node):
if curr_node is not None:
self.preShow(curr_node.getLeft())
self.preShow(curr_node.getRight())
print(curr_node.getLabel(), end=" ")
def getRoot(self): def getRoot(self):
return self.root return self.root
t = AVL() t = AVL()
t.insert(11) # t.insert(1)
t.insert(14) # t.preShow(t.getRoot())
t.insert(3) # print("\n")
t.insert(4) # t.preorden(t.getRoot())
t.insert(5) # print("\n")
t.insert(6) # t.insert(2)
t.insert(7) # t.preShow(t.getRoot())
t.insert(8) # print("\n")
t.insert(9) # t.preorden(t.getRoot())
t.insert(10) # print("\n")
# t.insert(3)
# t.preShow(t.getRoot())
# print("\n")
# t.preorden(t.getRoot())
# print("\n")
# print(t.getRoot().getHeight())
# print(t.getRoot().getRight().getHeight())
t.insert(1) t.insert(1)
t.insert(12)
t.insert(13)
t.insert(2) t.insert(2)
t.insert(15) t.insert(3)
t.preShow(t.getRoot()) # t.insert(4)
# t.preShow(t.getRoot())
# print("\n")
# t.preorden(t.getRoot())
# print("\n")
# t.insert(5)
# t.preShow(t.getRoot())
# print("\n")
# t.preorden(t.getRoot())
# print("\n")
# t.insert(6)
# t.preShow(t.getRoot())
# print("\n")
# t.preorden(t.getRoot())
# print("\n")
# t.insert(7)
# t.preShow(t.getRoot())
# print("\n")
# t.preorden(t.getRoot())
# print("\n")
# t.insert(8)
# t.preShow(t.getRoot())
# print("\n")
# t.preorden(t.getRoot())
# print("\n")
# t.insert(9)
# t.preShow(t.getRoot())
# print("\n")
# t.preorden(t.getRoot())
# print("\n")
# t.insert(10)
# t.preShow(t.getRoot())
# print("\n")
# t.preorden(t.getRoot())
# print("\n")
# t.insert(11)
# t.preShow(t.getRoot())
# print("\n")
# t.preorden(t.getRoot())
# print("\n")
# t.insert(12)
# t.preShow(t.getRoot())
# print("\n")
# t.preorden(t.getRoot())
# print("\n")
# t.insert(13)
# t.preShow(t.getRoot())
# print("\n")
# t.preorden(t.getRoot())
# print("\n")
# t.insert(14)
# t.preShow(t.getRoot())
# print("\n")
# t.preorden(t.getRoot())
# print("\n")
# t.insert(15)
# t.preShow(t.getRoot())
# print("\n")
# t.preorden(t.getRoot())