Cod sursa(job #424044)

Utilizator gabipurcaruGabi Purcaru gabipurcaru Data 24 martie 2010 16:04:43
Problema Trie Scor 40
Compilator cpp Status done
Runda Arhiva educationala Marime 1.2 kb
#include <fstream>
#include <cstring>
using namespace std;

ifstream in("trie.in");
ofstream out("trie.out");

struct Trie
{
    int nr, pref;
    Trie *fii[26], *tata;
};

Trie *t, *it, *it2;
int n, op, val, i, l, len;
char s[22],*p;

int main()
{
    t = new Trie;
    while(in>>op>>s)
    {
        if(!strcmp(s, "latitudine"))
            p=p;
        p = s;
        it = t;
        l = 0;len=0;
        while(p[0] != '\0')
        {
            if(it->fii[p[0] - 'a'] == NULL)
                it->fii[p[0] - 'a'] = new Trie, it->fii[p[0] - 'a']->tata = it,
                        it = it->fii[p[0] - 'a'];
            else
                it = it->fii[p[0] - 'a'];
            if(op == 0)
                ++ it->pref;
            else if(op == 1 && it->nr > 0)
                -- it->pref;
            else if(op == 3 && it->pref)
                l = p-s+1;


            p++;
        }
        if(op == 0)
            it->nr ++;
        else if(op == 1 && it->nr > 0)
        {
            it->nr --;
            while(it->pref <= 1)
                it2 = it->tata, delete it, it = it2;
        }
        else if(op == 2)
            out<<it->nr<<'\n';
        else if(op == 3)
            out<<l<<'\n';

    }
    return 0;
}