154 lines
3.5 KiB
C++
154 lines
3.5 KiB
C++
#include <iostream>
|
|
#include <fstream>
|
|
#include "Trie.h"
|
|
#include <vector>
|
|
#include "illegal_exception.h"
|
|
|
|
using namespace std;
|
|
|
|
const string FILE_NAME = "corpus.txt";
|
|
|
|
void PrintVector(vector<string> *out)
|
|
{
|
|
for (int i = 0; i < out->size(); i++)
|
|
{
|
|
cout << (*out)[i];
|
|
if (i < out->size() - 1)
|
|
cout << " ";
|
|
}
|
|
}
|
|
|
|
int main()
|
|
{
|
|
string cmd;
|
|
string addWord;
|
|
ifstream fin(FILE_NAME);
|
|
Trie *trie = new Trie();
|
|
|
|
while (cin >> cmd)
|
|
{
|
|
if (cmd == "load")
|
|
{
|
|
if (!fin.is_open())
|
|
fin.open(FILE_NAME);
|
|
|
|
while (fin >> addWord)
|
|
{
|
|
trie->Insert(addWord);
|
|
}
|
|
|
|
fin.close();
|
|
cout << "success" << endl;
|
|
}
|
|
else if (cmd == "i")
|
|
{
|
|
string word;
|
|
cin >> word;
|
|
try
|
|
{
|
|
if (trie->Insert(word))
|
|
cout << "success" << endl;
|
|
else
|
|
cout << "failure" << endl;
|
|
}
|
|
catch (illegal_exception e)
|
|
{
|
|
cout << "illegal argument" << endl;
|
|
}
|
|
}
|
|
else if (cmd == "e")
|
|
{
|
|
string word;
|
|
cin >> word;
|
|
|
|
try
|
|
{
|
|
if (trie->Remove(word))
|
|
cout << "success" << endl;
|
|
else
|
|
cout << "failure" << endl;
|
|
}
|
|
catch (illegal_exception e)
|
|
{
|
|
cout << "illegal argument" << endl;
|
|
}
|
|
}
|
|
else if (cmd == "clear")
|
|
{
|
|
trie->Clear();
|
|
trie = new Trie();
|
|
cout << "success" << endl;
|
|
}
|
|
else if (cmd == "size")
|
|
{
|
|
cout << "number of words is " << trie->Size() << endl;
|
|
}
|
|
else if (cmd == "empty")
|
|
{
|
|
if (trie->IsEmpty())
|
|
cout << "empty 1" << endl;
|
|
else
|
|
cout << "empty 0" << endl;
|
|
}
|
|
else if (cmd == "spellcheck")
|
|
{
|
|
string word;
|
|
cin >> word;
|
|
|
|
vector<string> out;
|
|
|
|
bool result = trie->SpellCheck(&out, word, "");
|
|
|
|
if (out.empty() && result)
|
|
{
|
|
cout << "correct" << endl;
|
|
}
|
|
else if (!result)
|
|
{
|
|
cout << endl;
|
|
}
|
|
else
|
|
{
|
|
PrintVector(&out);
|
|
cout << endl;
|
|
}
|
|
}
|
|
else if (cmd == "p")
|
|
{
|
|
vector<string> out;
|
|
trie->DepthFirstSearch(&out, "");
|
|
|
|
PrintVector(&out);
|
|
|
|
if (out.size() > 0)
|
|
cout << endl;
|
|
|
|
}
|
|
|
|
else if (cmd == "c")
|
|
{
|
|
string prefix;
|
|
cin >> prefix;
|
|
|
|
try
|
|
{
|
|
int count = trie->CountSuffixes(prefix);
|
|
|
|
if (count != -1 && count != 0)
|
|
cout << "count is " << count << endl;
|
|
else
|
|
cout << "not found" << endl;
|
|
}
|
|
catch (illegal_exception e)
|
|
{
|
|
cout << "illegal argument" << endl;
|
|
}
|
|
}
|
|
else if (cmd == "exit")
|
|
{
|
|
break;
|
|
}
|
|
}
|
|
trie->Clear();
|
|
}
|