Pagini recente » Cod sursa (job #459456) | Cod sursa (job #3000446) | Cod sursa (job #2123819) | Cod sursa (job #2431966) | Cod sursa (job #1560228)
#include <fstream>
#include <string>
#include <tuple>
#include <unordered_map>
using InstructionWordPair = std::tuple<char, std::string>;
struct TrieNode
{
TrieNode()
{
for (int i = 0; i < 26; ++i)
{
Children[i] = nullptr;
}
}
int OccurrenceCount = 0;
int ChildCount = 0;
TrieNode* Children[26];
};
TrieNode* root = new TrieNode();
InstructionWordPair LineToInstructionAndWord(const std::string& line)
{
char instruction = line[0];
std::string word = line.substr(2);
return std::make_tuple(instruction, word);
}
namespace detail
{
int CharToChildIndex(char c)
{
return c - 'a';
}
bool Delete(TrieNode* node, const std::string& word, int indexInWord)
{
if (indexInWord == word.length())
{
node->OccurrenceCount--;
}
else
{
int childIndex = CharToChildIndex(word[indexInWord]);
auto child = node->Children[childIndex];
if (Delete(child, word, indexInWord + 1))
{
delete child;
node->Children[childIndex] = nullptr;
node->ChildCount--;
}
}
return node->OccurrenceCount <= 0 && node->ChildCount <= 0;
}
int CountLongestCommonPrefix(const TrieNode* node, const std::string& word, int indexInWord, int lengthSoFar)
{
if (indexInWord == word.length())
{
return lengthSoFar;
}
int childIndex = CharToChildIndex(word[indexInWord]);
auto child = node->Children[childIndex];
if (child != nullptr)
{
return CountLongestCommonPrefix(child, word, indexInWord + 1, lengthSoFar + 1);
}
return lengthSoFar;
}
void AddWordOccurrence(TrieNode* node, const std::string& word, int indexInWord)
{
if (indexInWord == word.length())
{
node->OccurrenceCount++;
}
else
{
int childIndex = CharToChildIndex(word[indexInWord]);
auto child = node->Children[childIndex];
if (child == nullptr)
{
auto newNode = new TrieNode();
node->Children[childIndex] = newNode;
node->ChildCount++;
AddWordOccurrence(newNode, word, indexInWord + 1);
}
else
{
AddWordOccurrence(child, word, indexInWord + 1);
}
}
}
int CountOccurrences(TrieNode* node, const std::string& word, int indexInWord)
{
if (indexInWord == word.length())
{
return node->OccurrenceCount;
}
int childIndex = CharToChildIndex(word[indexInWord]);
auto child = node->Children[childIndex];
if (child != nullptr)
{
return CountOccurrences(child, word, indexInWord + 1);
}
return 0;
}
}
void AddWordOccurrence(const std::string& word)
{
detail::AddWordOccurrence(root, word, 0);
}
void DeleteWordOccurrence(const std::string& word)
{
detail::Delete(root, word, 0);
}
int CountOccurrences(const std::string& word)
{
return detail::CountOccurrences(root, word, 0);
}
int CountLongestCommonPrefix(const std::string& word)
{
return detail::CountLongestCommonPrefix(root, word, 0, 0);
}
void Execute(const InstructionWordPair& instructionAndWord, std::ostream& out)
{
const char instruction = std::get<0>(instructionAndWord);
const std::string word = std::get<1>(instructionAndWord);
switch (instruction)
{
case('0') :
AddWordOccurrence(word);
break;
case('1') :
DeleteWordOccurrence(word);
break;
case('2') :
out << CountOccurrences(word) << std::endl;
break;
case('3') :
out << CountLongestCommonPrefix(word) << std::endl;
break;
default:
break;
}
}
int main()
{
std::ifstream in("trie.in");
std::ofstream out("trie.out");
std::string line;
while (std::getline(in, line))
{
auto instructionAndWord = LineToInstructionAndWord(line);
Execute(instructionAndWord, out);
}
out.close();
in.close();
}