Searching the String ZOJ

Searching the String

 ZOJ - 3228

题意：两种查询，普通的就是ac自动机模板，另外一种要求没有发生覆盖。

处理方式是判断一下如果当前匹配到的位置(u)减去上一次匹配到的位置（ed[u]）大于该字符的深度（deep[u])，就加1。

#include <bits/stdc++.h>
using namespace std;
#define CLR(m,a) memset(m,a,sizeof(m))

const int maxnode=5*1e5+10;
const int sigma=26;

struct AC{
    int ch[maxnode][sigma],f[maxnode],last[maxnode];
    int val[maxnode],deep[maxnode];
    int cnt[maxnode][2];
    int ed[maxnode];
    int sz;

    void init(){
        CLR(ch[0],0);
        sz=1;
        CLR(ed,-1);
        CLR(val,0);
        CLR(cnt,0);
    }

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

    int inser(char *s){
        int u=0,n=strlen(s);
        for(int i=0;i<n;i++){
            int c=idx(s[i]);
            if(!ch[u][c]){
                CLR(ch[sz],0);
                deep[sz]=i+1;   //
                ch[u][c]=sz++;
            }
            u=ch[u][c];
        }
        val[u]=1;
        return u;
    }

    void getfail(){
        queue<int> q;
        f[0]=0;
        last[0]=0;
        for(int c=0;c<sigma;c++){
            int u=ch[0][c];
            if(u){
                f[u]=0;
                last[u]=0;
                q.push(u);
            }
        }
        while(!q.empty()){
            int r=q.front();
            q.pop();
            for(int c=0;c<sigma;c++){
                int u=ch[r][c];
                if(!u) continue;
                q.push(u);
                int v=f[r];
                while(v&&!ch[v][c]) v=f[v];
                f[u]=ch[v][c];
                last[u]=val[f[u]]?f[u]:last[f[u]];
            }
        }
    }
    void print(int u,int i){
        if(val[u]){
            cnt[u][0]++;
            if(i-ed[u]>=deep[u]){
                cnt[u][1]++;
                ed[u]=i;
            }
            print(last[u],i);
        }
    }
    void query(char *s){
        int u=0,n=strlen(s);
        for(int i=0;i<n;i++){
            int c=idx(s[i]);
            while(u&&!ch[u][c]) u=f[u];
            u=ch[u][c];
            if(val[u]) print(u,i);
            else if(last[u]) print(last[u],i);
        }
    }
};
const int maxn=1e5+10;
int pos[maxn],c[maxn];
char s[maxn],p[10];
AC ac;
int main(){
    int kase=0;
    while(scanf("%s",s)!=EOF){
        ac.init();
        printf("Case %d
",++kase);
        int n;
        scanf("%d",&n);
        for(int i=0;i<n;i++){
            scanf("%d%s",&c[i],p);
            pos[i]=ac.inser(p);
        }
        ac.getfail();
        ac.query(s);
        for(int i=0;i<n;i++){
            printf("%d
",ac.cnt[pos[i]][c[i]]);
        }
        puts("");
    }
    return 0;
}