48 lines
882 B
Python
48 lines
882 B
Python
|
from comparator import Comparator
|
||
|
|
from avl import AVLTree
|
||
|
|
from wavl import WAVLTree
|
||
|
|
|
||
|
|
import logging
|
||
|
|
import random
|
||
|
|
|
||
|
|
|
||
|
|
def main():
|
||
|
|
logging.basicConfig(level=logging.DEBUG)
|
||
|
|
|
||
|
|
t = Comparator(AVLTree(), WAVLTree())
|
||
|
|
# t = WAVLTree()
|
||
|
|
|
||
|
|
values = []
|
||
|
|
for _ in range(42):
|
||
|
|
x = random.randint(1, 100)
|
||
|
|
while x in values:
|
||
|
|
x = random.randint(1, 100)
|
||
|
|
|
||
|
|
values.append(x)
|
||
|
|
t.insert(x)
|
||
|
|
|
||
|
|
for _ in range(5):
|
||
|
|
x = random.choice(values)
|
||
|
|
values.remove(x)
|
||
|
|
t.delete(x)
|
||
|
|
|
||
|
|
for _ in range(42):
|
||
|
|
x = random.randint(1, 100)
|
||
|
|
while x in values:
|
||
|
|
x = random.randint(1, 100)
|
||
|
|
|
||
|
|
values.append(x)
|
||
|
|
t.insert(x)
|
||
|
|
|
||
|
|
for _ in range(5):
|
||
|
|
x = random.choice(values)
|
||
|
|
values.remove(x)
|
||
|
|
t.delete(x)
|
||
|
|
|
||
|
|
print(t)
|
||
|
|
# print(t.are_equal)
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
main()
|