Cod sursa(job #104528)

Utilizator filipbFilip Cristian Buruiana filipb Data 16 noiembrie 2007 12:00:14
Problema Abc2 Scor 0
Compilator cpp Status done
Runda Happy Coding 2007 Marime 1.92 kb
#include <stdio.h>
#include <string.h>

#define MaxNodes 1000005
#define NMax 10000005

char S[NMax], cuv[25], F[MaxNodes];
int trie[MaxNodes][3], pi[MaxNodes], A[MaxNodes][3], nodes = 1;
int q[MaxNodes], cnt;

void build_automaton(void)
{
    int i, c, p, ql, qr, f = 0;

    for (q[ql=qr=0]=1, pi[1] = 0; qr <= ql; qr++)
        for (c = 0; c < 3; c++)
            if (trie[q[qr]][c])
            {            
                q[++ql] = trie[q[qr]][c];
                
                for (p = pi[q[qr]]; p && !trie[p][c]; p = pi[p]);
                
                if (!p) pi[q[ql]] = 1;
                else pi[q[ql]] = trie[p][c];                
            }

    for (i = 1; i <= nodes; i++)
        for (c = 0; c < 3; c++)
            if (trie[i][c])
                A[i][c] = trie[i][c];
            else
            {
                for (p = pi[i]; p && !trie[p][c]; p = pi[p]);

                if (!p) A[i][c] = 1;
                else A[i][c] = trie[p][c];
            }
    
}

int main(void)
{
    int i, nod, L = -1, len;
    
    freopen("abc2.in", "r", stdin);
    freopen("abc2.out", "w", stdout);

    memset(trie, 0, sizeof(trie));
    
    fgets(S, sizeof(S), stdin);
    while (fgets(cuv, sizeof(cuv), stdin) != NULL)
    {
        if (L == -1)
            for (L = 0; cuv[L] >= 'a' && cuv[L] <= 'c'; L++);
        
        for (i = 0, nod = 1; i < L; i++)
        {
            if (cuv[i] < 'a' || cuv[i] >'c') break;
            
            if (!trie[nod][cuv[i]-'a'])
                trie[nod][cuv[i]-'a'] = (++nodes);
            nod = trie[nod][cuv[i]-'a'];
        }
        
        F[nod] = 1;
    }

    build_automaton();

    for (i = 0, nod = 1, len = strlen(S); i < len; i++)
    {
        if (S[i] < 'a' || S[i] > 'c') break;

        nod = A[nod][S[i]-'a'];
        cnt += (F[nod] == 1);
    }

    printf("%d\n", cnt);

    return 0;
}