void TestAVLTree(int argc, char **argv)
{
int array[TREE_SIZE];
int seed;
bool IsInsert;
if (argc == 2)
seed = atoi(argv[1]);
else
seed = time(0) *257%32768;
SimpleCompareNodesAlgorithm<int, int> CompareAlgorithm;
for (int iteration = 1; iteration <= N_ITERATIONS; iteration++)
{
AVLTree<int, int> *tree;
int i;
printf("Iteration %4d/%4d: seed=%5d", iteration, N_ITERATIONS, seed);
fflush(stdout);
srand(seed++);
for (i = 0; i < TREE_SIZE; i++)
array[i] = i;
Shuffle(array, TREE_SIZE);
tree = new AVLTree<int, int>(&CompareAlgorithm);
for (i = 0; i< TREE_SIZE; i++)
tree->Search(&(array[i]), &IsInsert);
// if (tree->Verify(&IsDone))
// return;
for(i = 0; i < TREE_SIZE; i++)
{
tree->Delete(&(array[i]));
// if (tree->Verify(&IsDone))
// return;
}
printf("\ttest good.\n");
fflush(stdout);
delete tree;
}
}
int main()
{
AVLTree<int> itree;
std::cout << "Insert 0 9 4 6 -4 -2 -1\n";
itree.Insert(0);
itree.Insert(9);
itree.Insert(4);
itree.Insert(6);
itree.Insert(-4);
itree.Insert(-2);
itree.Insert(-1);
std::cout << "InOrder traverse:\n";
itree.InOrderTraverse();
std::cout << "PreOrder traverse:\n";
itree.PreOrderTraverse();
std::cout << "Deleted 9:\n";
itree.Delete(9);
std::cout << "InOrder:\n";
itree.InOrderTraverse();
std::cout << "PreOrder:\n";
itree.PreOrderTraverse();
std::cout << "Deleted 4:\n";
itree.Delete(4);
std::cout << "InOrder:\n";
itree.InOrderTraverse();
std::cout << "PreOrder:\n";
itree.PreOrderTraverse();
std::cout << "Deleted 0:\n";
itree.Delete(0);
std::cout << "InOrder:\n";
itree.InOrderTraverse();
std::cout << "PreOrder:\n";
itree.PreOrderTraverse();
std::cout << "Deleted -2:\n";
itree.Delete(-2);
std::cout << "InOrder:\n";
itree.InOrderTraverse();
std::cout << "PreOrder:\n";
itree.PreOrderTraverse();
std::cout << "Deleted 6:\n";
itree.Delete(6);
std::cout << "InOrder:\n";
itree.InOrderTraverse();
std::cout << "PreOrder:\n";
itree.PreOrderTraverse();
//-----------------------------------------------------------------------------------------
AVLTree<std::string> strTree;
std::cout << "Inserted 5 string: rensehngqiang shibojun lijin wangzaibing zhaogang\n";
strTree.Insert("renshengqiang");
strTree.Insert("shibojun");
strTree.Insert("lijin");
strTree.Insert("wangzaibing");
strTree.Insert("zhaogang");
std::cout << "InOrder traverse:\n";
strTree.InOrderTraverse();
std::cout << "PreOrder traverse:\n";
strTree.PreOrderTraverse();
std::cout << "Look for lijin:\n";
AVLTreeNode<std::string> *pNode = strTree.Search("lijin");
if(pNode != 0) std::cout << pNode->element << std::endl;
std::cout << "Look for huangliangyu\n";
pNode = strTree.Search("huangliangyu");
if(pNode != 0) std::cout << pNode->element << std::endl;
return 0;
}