zoj -- 3228 Searching the String(AC自动机)

给出一个字符串。有n个询问，0 string表示string在字符串中出现多少次，可以重叠。1 string表示string在字符串中出现多少次，string不能重叠。

http://acm.zju.edu.cn/onlinejudge/showProblem.do?problemCode=3228

维护一个last数组，记录该字符串上次出现的位置，根据这个来判断有没有重叠。

//#pragma comment(linker,"/STACK:1024000000,1024000000")
#include <map>
#include <set>
#include <cmath>
#include <queue>
#include <stack>
#include <cstdio>
#include <string>
#include <vector>
#include <cstring>
#include <iostream>
#include <algorithm>
using namespace std;

typedef double DB;
typedef long long ll;
typedef unsigned long long ull;
typedef pair<int, int> PII;

#define pb push_back
#define MP make_pair
#define lson l, m, rt << 1
#define rson m + 1, r, rt << 1 | 1

const double eps = 1e-6;
const int inf = 0x3f3f3f3f;
const int mod = 100000;
const int maxn = 600000 + 10;

char str[100010], ss[100];
int n, op[100010], pos[100010], cas = 0;
int tot[maxn][2], last[maxn];
int d[maxn];

const int max_chd = 26;///字母表大小
struct AC_automata{
    int chd[maxn][max_chd], fail[maxn], cnt[maxn];///trie中的chd数组，fail指针，节点标记（是多少个串的结尾）
    int root, sz;///根节点，trie树大小
    int newnode(){///建立新的节点，返回节点编号
        for(int i=0; i<max_chd; i++) chd[sz][i] = -1;
        d[sz] = 0;
        cnt[sz++] = 0;
        return sz - 1;
    }
    void init(){
        sz = 0;
        root = newnode();
    }
    int ID(char ch){
        return ch - 'a';
    }
    int Insert(char * str){///建trie树
        int cur = root;
        int len = strlen(str);
        for(int i=0; i<len; i++){
            if(chd[cur][ID(str[i])] == -1)
                chd[cur][ID(str[i])] = newnode(), d[chd[cur][ID(str[i])]] = i + 1;
            cur = chd[cur][ID(str[i])];
        }
        cnt[cur]++;
        return cur;
    }
    void build(){///构建fail指针
        queue<int> Q;
        while(!Q.empty()) Q.pop();
        fail[root] = root;
        for(int i=0; i<max_chd; i++){
            if(chd[root][i] == -1)
                chd[root][i] = root;
            else{
                fail[chd[root][i]] = root;
                Q.push(chd[root][i]);
            }
        }
        while(!Q.empty()){
            int now = Q.front(); Q.pop();
            for(int i=0; i<max_chd; i++){
                if(chd[now][i] == -1)
                    chd[now][i] = chd[fail[now]][i];
                else {
                    fail[chd[now][i]] = chd[fail[now]][i];
                    Q.push(chd[now][i]);
                }
            }
        }
    }
    void Query(char * str){
        memset(tot, 0, sizeof(tot));
        memset(last, -1, sizeof(last));
        int now = root;
        int len = strlen(str);
        for(int i=0; i<len; i++){
            now = chd[now][ID(str[i])];
            int tmp = now;
            while(tmp != root){
                if(cnt[tmp]){
                    tot[tmp][0]++;
                    if(i - last[tmp] >= d[tmp]) tot[tmp][1]++, last[tmp] = i;
                }
                tmp = fail[tmp];
            }
        }
        for(int i=0; i<n; i++){
            printf("%d\n", tot[pos[i]][op[i]]);
        }
    }
}AC;


int main(){
    while(scanf("%s", str) == 1){
        scanf("%d", &n);
        AC.init();
        for(int i=0; i<n; i++){
            scanf("%d%s", &op[i], ss);
            pos[i] = AC.Insert(ss);
        }
        AC.build();
        printf("Case %d\n", ++cas);
        AC.Query(str);
        puts("");
    }
    return 0;
}