【TJOI2013】 单词

【题目链接】

点击打开链接

【算法】

AC自动机+递推

【代码】

#include<bits/stdc++.h>
using namespace std;
#define MAXN 200
const int MAXL = 1e6+;

int i,N;
int pos[MAXN+];
char s[MAXN+][MAXL];

template <typename T> inline void read(T &x) {
    int f = ; x = ;
    char c = getchar();
    for (; !isdigit(c); c = getchar()) { if (c == '-') f = -f; }
    for (; isdigit(c); c = getchar()) x = x *  + c - '';
    x *= f;
}

template <typename T> inline void write(T x) {
    if (x < ) { putchar('-'); x = -x; }
    if (x > ) write(x/);
    putchar(x%+'');
}

template <typename T> inline void writeln(T x) {
    write(x);
    puts("");
}

struct AC_Automation {
    int tot;
    struct Trie {
        int sum,fail;
        int next[];
    } trie[MAXL];
    inline void Insert(int id,char *s) {
        int i,len,x,root=;
        len = strlen(s);
        for (i = ; i < len; i++) {
            x = s[i] - 'a';
            if (!trie[root].next[x]) trie[root].next[x] = ++tot;
            root = trie[root].next[x];
            ++trie[root].sum;
        }
        pos[id] = root;
    }
    inline void rebuild() {
        int i,root,tmp,l=,r=;
        static int q[MAXL];
        q[] = ;
        trie[].fail = -;
        while (l <= r) {
            root = q[l]; l++;
            for (i = ; i < ; i++) {
                if (trie[root].next[i]) {
                    if (!root)
                        trie[trie[root].next[i]].fail = ;
                    else {
                        tmp = trie[root].fail;
                        while (tmp != -) {
                            if (trie[tmp].next[i]) {
                                trie[trie[root].next[i]].fail = trie[tmp].next[i];
                                break;
                            }
                            tmp = trie[tmp].fail;
                        }
                        if (tmp == -) trie[trie[root].next[i]].fail = ;
                    }
                    q[++r] = trie[root].next[i];
                }
            }
        }
        for (i = r; i >= ; i--) trie[trie[q[i]].fail].sum += trie[q[i]].sum;
    }
    inline void output() {
        int i;
        for (i = ; i <= N; i++) writeln(trie[pos[i]].sum);
    }
} ACAM;

int main() {

    read(N);
    for (i = ; i <= N; i++) {
        gets(s[i]);
        ACAM.Insert(i,s[i]);
    }
    ACAM.rebuild();
    ACAM.output();

    return ;
}