Cod sursa(job #998166)

Utilizator Cosmin1490Balan Radu Cosmin Cosmin1490 Data 15 septembrie 2013 23:11:21
Problema Trie Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 2.18 kb
#include <vector>
#include <list>
#include <map>
#include <set>
#include <deque>
#include <queue>
#include <stack>
#include <bitset>
#include <algorithm>
#include <functional>
#include <numeric>
#include <utility>
#include <sstream>
#include <iostream>
#include <iomanip>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <cctype>
#include <string>
#include <cstring>
#include <cstdio>
#include <cmath>
#include <cstdlib>
#include <ctime>
#include <fstream>
using namespace std;

const string file = "trie";

const string infile = file + ".in";
const string outfile = file + ".out";

const int INF = 0x3f3f3f3f;

struct Nod
{
	int fii;
	int nr;
	Nod* n[26];

	Nod()
	{
		fii = 0;
		nr = 0;
		for(int i = 0; i < 26; i++)
		{
			n[i] = NULL;
		}
	}

};

Nod* root = new Nod();


void insert(Nod* r, string& s, int p)
{
	Nod* c = r;
	for(int i = 0; i < (int)s.size(); i++)
	{
		if(c->n[s[i] - 'a'] == NULL)
		{
			c->n[s[i] - 'a'] = new Nod();
			c->fii ++;
		}
		c = c->n[s[i] - 'a'];
	}
	c->nr++;
}


bool erase(Nod* r, string& s, int p)
{
	if(p == s.size())
	{
		r->nr--;
	}
	else if(erase(r->n[s[p] - 'a'], s, p+1))
	{
		r->fii--;
		r->n[s[p] - 'a'] = NULL;
	}

	if(r->nr == 0 && r->fii == 0 && r != root)
	{
		delete r;
		return true;
	}

	return false;
}


int wordCount(Nod* r, string& s, int p)
{
	Nod* c = r;
	for(unsigned i = 0; i < s.size(); i++)
	{
		if(c->n[s[i] - 'a'] == NULL)
		{
			return 0;
		}
		c = c->n[s[i] - 'a'];
	}
	return c->nr;
}

int wordPrefix(Nod* r, string& s, int p)
{
	Nod* c = r;
	for(unsigned i = 0; i < s.size(); i++)
	{
		if(c->n[s[i] - 'a'] == NULL)
		{
			return i;
		}
		c = c->n[s[i] - 'a'];
	}
	return s.size();
}

int main()
{
	fstream fout(outfile.c_str(), ios::out);
	fstream fin(infile.c_str(), ios::in);
	int op;
	string s;

	while(fin >> op && fin >> s)
	{

		switch(op)
		{
		case 0:
			insert(root, s, 0);
			break;
		case 1:
			erase(root, s, 0);
			break;
		case 2:
			fout << wordCount(root, s, 0) << "\n";
			break;
		case 3:
			fout << wordPrefix(root, s, 0) << "\n";
			break;
		}

	}


	fin.close();
	fout.close();
}