HDU 3065 病毒侵袭持续中

询问每个模式串在文本传中出现的次数。

文本串中出现的字符不一定都是大写字母，只需要在匹配的时候，对文本串进行特殊处理，将连续的大写字母段当成合法的一个文本串即可。

然后……就是简单的统计了。

 

#include <cstdio>

#include <algorithm>

#include <vector>

#include <queue>

#include <cstring>

#include <string>

using namespace std;

int n, cnt[1010];

char s[1010][55], t[2000010], a[2000010];



struct AC_Automata {

    #define N 60003

    #define M 26

    int ch[N][M], f[N], val[N], last[N], sz;



    void clear() { sz = 1; memset(ch[0], 0, sizeof(ch[0])); }

    int idx(char c) { return c-'A'; }



    void insert(char s[], int v) {

        int u = 0;

        for (int i=0; s[i]; i++) {

            int c = idx(s[i]);

            if (!ch[u][c]) {

                memset(ch[sz], 0, sizeof(ch[sz]));

                val[sz] = 0;

                ch[u][c] = sz++;

            }

            u = ch[u][c];

        }

        val[u] = v;

    }

    void build() {

        queue<int> q;

        f[0] = 0;

        for (int c=0; c<M; c++) {

            int u = ch[0][c];

            if (u) { f[u] = last[u] = 0; q.push(u); }

        }

        while (!q.empty()) {

            int r = q.front(); q.pop();

            for (int c=0; c<M; c++) {

                int u = ch[r][c];

                if (!u) { ch[r][c] = ch[f[r]][c]; continue; }

                q.push(u);

                f[u] = ch[f[r]][c];

                last[u] = val[f[u]] ? f[u] : last[f[u]];

            }

        }

    }

    void find(char s[]) {

        int j = 0;

        for (int i=0; s[i]; i++) {

            int c = idx(s[i]);

            j = ch[j][c];

            if (val[j]) print(j);

            else if (last[j]) print(last[j]);

        }

    }

    void print(int j) {

        if (j) {

            cnt[val[j]]++;

            print(last[j]);

        }

    }

} ac;





int main() {

#ifndef ONLINE_JUDGE

    freopen("in.txt", "r", stdin);

#endif

    while (scanf(" %d", &n) == 1) {

        ac.clear();

        memset(cnt, 0, sizeof(cnt));



        getchar();

        for (int i=1; i<=n; i++) {

            gets(s[i]); ac.insert(s[i], i);

        }

        ac.build();



        gets(t);

        int j = 0;

        bool flag = false;

        for (int i=0; t[i]; i++) {

            if (t[i] <= 'Z' && t[i] >= 'A') { flag = true; a[j++] = t[i]; }

            else if (flag) {

                a[j] = 0; j = 0;

                flag = false;

                ac.find(a);

            }

        }

        if (flag) {

            a[j] = 0; j = 0;

            flag = false; ac.find(a);

        }



        for (int i=1; i<=n; i++)

            if (cnt[i]) printf("%s: %d\n", s[i], cnt[i]);

    }

    return 0;

}