dm/school
1
0
Fork 0
Code Issues Pull Requests Activity

adding cs235

Browse Source
This commit is contained in:
dm
2016-04-06 20:46:10 -07:00
parent 8e52ce1982
commit cf99ec6565
178 changed files with 13079 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
cs235/final/main.cpp
+235
View File
@@ -0,0 +1,235 @@
#include <iostream>
#include <sstream>
#include <string>
#include <time.h>
#include <stdlib.h>
#include "RedBlackTreeFactory.h"
#include "TwoThreeTreeFactory.h"
#include "br_tree.h"
/* This main is provided to help you test your code. Be aware that the TAs will use a separate
* test driver that will only test your red black tree code. You may modify this code if you wish,
* but do not try to modify the main to get your own code to work.
*/
using namespace std;
int getOption(int max, int min);
string getString();
void exitPause();
RedBlackTreeInterface * redBlackTree;
TwoThreeTreeInterface * twoThreeTree;
int main()
{
srand(time(NULL));
br_tree b;
redBlackTree = RedBlackTreeFactory::getRedBlackTree();
if(redBlackTree == NULL) {
cout << endl;
cout << "Factory getRedBlackTree() returned NULL!\nExiting Test Driver (press enter to exit)" << endl;
exitPause();
return 0;
}
// twoThreeTree = TwoThreeTreeFactory::getTwoThreeTree();
cout << "Final Exam Beginning" << endl;
while(1)
{
int choice = 0;
if(twoThreeTree == NULL) {
cout<< "Please select an option:"
<< "\n1. Add a single word to the tree"
<< "\n2. Add a list of words to the tree"
<< "\n3. Remove a single word from the tree"
<< "\n4. Print the tree"
<< "\n5. Quit Program" << endl;
choice = getOption(1,5);
}
else {
cout<< "Please select an option:"
<< "\n1. Add a single word to the red black tree"
<< "\n2. Add a list of words to the red black tree"
<< "\n3. Remove a single word from the red black tree"
<< "\n4. Print the red black tree"
<< "\n5. Add a single word to the two three tree"
<< "\n6. Add a list of words to the two three tree"
<< "\n7. Remove a single word from the two three tree"
<< "\n8. Print the two three tree"
<< "\n9. Quit Program" << endl;
choice = getOption(1,9);
}
if(choice == 1)
{
//Insert the string that the user passes to the program into the tree
for(int i = 0; i < 50000; i++) {
stringstream s;
int rand_n = rand() % 50000;
s << rand_n;
b.add(s.str());
}
cout << b.root << endl;
//cout<< "Enter the word to add to the tree (alphabetical characters, no punctuation marks, no spaces)."<<endl;
//string insertToTree;
//insertToTree = getString();
//redBlackTree->add(insertToTree);
}
else if(choice == 2)
{
//Insert the string that the user passes to the program into the tree
cout<< "Enter the string to add to the tree."
<< "\n(alphabetical characters, no punctuation marks, and with white space as the delimiter separating words)" << endl;
string insertToTree;
insertToTree = getString();
redBlackTree->addPhrase(insertToTree);
}
else if(choice == 3)
{
cout << "here" << endl;
while(b.root) {
cout << "ah" << endl;
stringstream s;
int rand_n = rand() % 50000;
s << rand_n;
b.remove(s.str());
}
//Remove the string that the user passes to the program into the tree
//cout << "Enter the word to remove from the tree (alphabetical characters, no punctuation marks, no spaces)."<<endl;
//string removeFromTree;
//removeFromTree = getString();
//redBlackTree->remove(removeFromTree);
}
else if(choice == 4)
{
cout << b.printTree();
//cout << "Your tree:" << endl;
//cout << "\t" << redBlackTree->printTree() << endl;
}
else if(choice == 5 && twoThreeTree == NULL)
{
break;
}
else if(choice == 5 && twoThreeTree != NULL)
{
//Insert the string that the user passes to the program into the tree
cout<< "Enter the word to add to the tree (alphabetical characters, no punctuation marks, no spaces)."<<endl;
string insertToTree;
insertToTree = getString();
twoThreeTree->add(insertToTree);
}
else if(choice == 6)
{
//Insert the string that the user passes to the program into the tree
cout<< "Enter the string to add to the tree."
<< "\n(alphabetical characters, no punctuation marks, and with white space as the delimiter separating words)" << endl;
string insertToTree;
insertToTree = getString();
twoThreeTree->addPhrase(insertToTree);
}
else if(choice == 7)
{
//Remove the string that the user passes to the program into the tree
cout << "Enter the word to remove from the tree (alphabetical characters, no punctuation marks, no spaces)."<<endl;
string removeFromTree;
removeFromTree = getString();
twoThreeTree->remove(removeFromTree);
}
else if(choice == 8)
{
cout << "Your tree:" << endl;
cout << "\t" << twoThreeTree->printTree() << endl;
}
else
{
break;
}
}
cout << "Thanks for Playing!" << endl;
exitPause();
return 0;
}
//-------------------------------------------------------------------------------------
int getOption(int min, int max)
{
int input = 0;
bool done = false;
while(!done)
{
input = 0;
cin >> input;
cin.ignore(1000,'\n');
if(cin.fail())
{
cin.clear();
cin.ignore(1000,'\n');
cout << "Error: Invalid" << endl;
}
else if(input < min || input > max)
{
cout << "Error: Invalid number" << endl;
}
else
{
done = true;
}
}
return input;
}
string getString()
{
string input = "";
bool done = false;
while(!done)
{
input = "";
getline(cin, input);
if(cin.fail())
{
cin.clear();
cin.ignore(1000,'\n');
cout << "Error: Invalid name" << endl;
}
else
{
done = true;
}
}
return input;
}
string getWord()
{
string input = "";
bool done = false;
while(!done)
{
input = "";
getline(cin, input);
if(cin.fail())
{
cin.clear();
cin.ignore(1000,'\n');
cout << "Error: Invalid name" << endl;
}
else
{
done = true;
}
}
return input;
}
void exitPause() {
#ifdef ON_WINDOWS
system("pause");
#else
#endif
}