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

finish AVL

Browse Source
This commit is contained in:
Rafael 2017-07-07 20:06:33 +02:00
parent d8a0afc550
commit 8384cbb879
1 changed files with 56 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

82
data_structures/AVL/AVL.py
View File

@ -39,7 +39,7 @@ class Node:
def setHeight(self, height): def setHeight(self, height):
self.height = height self.height = height
def getHeight(self, height): def getHeight(self):
return self.height return self.height
@ -72,38 +72,36 @@ class AVL:
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) dad_node.setHeight(dad_node.getHeight() + 1)
if (dad_node.getRight().getHeight() -
dad_node.getLeft.getHeight() > 1):
self.rebalance(dad_node)
else: else:
dad_node.setRight(node) dad_node.setRight(node)
dad_node.setHeight(dad_node.getHeight() + 1) dad_node.setHeight(dad_node.getHeight() + 1)
if (dad_node.getRight().getHeight() -
dad_node.getLeft.getHeight() > 1):
self.rebalance(dad_node)
break break
def rebalance(self, node): def rebalance(self, node):
if (node.getRight().getHeight() - height_right = 0
node.getLeft.getHeight() > 1): height_left = 0
if (node.getRight().getHeight() >
node.getLeft.getHeight()): if node.getRight() is not None:
pass height_right = node.getRight().getHeight()
if node.getLeft() is not None:
height_left = node.getLeft().getHeight()
if abs(height_left - height_right) > 1:
if height_left > height_right:
right_child = node.getRight()
if (right_child.getLeft().getHeight() >
right_child.getRight().getHeight()):
self.rotate_left(node)
else:
self.double_rotate_right(node)
else: else:
pass left_child = node.getLeft()
pass if (left_child.getLeft().getHeight() >
elif (node.getRight().getHeight() - left_child.getRight().getHeight()):
node.getLeft.getHeight() > 2): self.double_rotate_left(node)
if (node.getRight().getHeight() > else:
node.getLeft.getHeight()): self.rotate_right(node)
pass
else:
pass
pass
pass
def rotate_left(self, node): def rotate_left(self, node):
# TODO: is this pythonic enought? # TODO: is this pythonic enought?
@ -129,3 +127,35 @@ class AVL:
def double_rotate_right(self, node): def double_rotate_right(self, node):
self.rotate_left(node.getLeft().getLeft()) self.rotate_left(node.getLeft().getLeft())
self.rotate_right(node) self.rotate_right(node)
def empty(self):
if self.root is None:
return True
return False
def preShow(self, curr_node):
if curr_node is not None:
self.preShow(curr_node.getLeft())
print(curr_node.getLabel(), end=" ")
self.preShow(curr_node.getRight())
def getRoot(self):
return self.root
t = AVL()
t.insert(11)
t.insert(14)
t.insert(3)
t.insert(4)
t.insert(5)
t.insert(6)
t.insert(7)
t.insert(8)
t.insert(9)
t.insert(10)
t.insert(1)
t.insert(12)
t.insert(13)
t.insert(2)
t.insert(15)
t.preShow(t.getRoot())