hdu5853 （后缀自动机）

Problem Jong Hyok and String

题目大意

　　给你n个字符串，有q个询问。

　　定义set(s)={(i,j)} 表示 s在第i个字符串中出现，且末尾位置为j。

　　对于一个询问，求set(Qi)=set(t) ，t的数量。

　　（n,q<=10^5 , 字符串总长<=10^5）

解题分析

　　直接将n个串塞进一个后缀自动机里面。

　　对于一个询问串qi，找到其在后缀自动机中出现的位置j。

　　则答案为len[j] - len[fail[j]] 。 （具体为什么还需要好好斟酌一下）

参考程序

#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;

#define V 100008
int ans[V];
struct sam{
    int nt[V*2][26],f[V*2],a[V*2],rig[V*2];
    int last,sum,len;
    int p,q,np,nq;

    void clear(){
        memset(f,-1,sizeof(f));
        memset(a,0,sizeof(a));
        memset(nt,-1,sizeof(nt));
        last = sum = len=0;
    }
    void insert(int ch){
        len++;
        if (~nt[last][ch] && a[nt[last][ch]]==len){
            last=nt[last][ch]; return;
        }
        p=last; last=np=++sum; a[np]=a[p]+1; rig[np]=1;
        for (;~p && !~nt[p][ch];p=f[p]) nt[p][ch]=np;
        if (p==-1) f[np]=0;
        else
        {
            q=nt[p][ch];
            if (a[q]==a[p]+1) f[np]=q;
            else
            {
                nq=++sum; a[nq]=a[p]+1;
                memcpy(nt[nq],nt[q],sizeof(nt[q]));
                f[nq]=f[q];
                f[q]=f[np]=nq;
                for (;~p && nt[p][ch]==q;p=f[p]) nt[p][ch]=nq;
            }
        }
    }
    int work(char *s,int l){
        int now=0;
        for (int i=1;i<=l;i++){
            if (nt[now][s[i]-'a']==-1) return 0;
            now=nt[now][s[i]-'a'];  
        }
        return a[now]-a[f[now]];
    }
}sam;

char str[V];
void work(){
    sam.clear();
    int n,m;
    scanf("%d %d",&n,&m);
    for (int i=1;i<=n;i++){
        sam.len=sam.last=0;
        scanf("%s",str+1);
        int len=strlen(str+1);
        for (int j=1;j<=len;j++) sam.insert(str[j]-'a');
    }
    for (int i=1;i<=m;i++){
        scanf("%s",str+1);
        int len=strlen(str+1);
        printf("%d
",sam.work(str,len));   
    }
}

int main(){
    int T;
    scanf("%d",&T);
    for (int t=1;t<=T;t++){
        printf("Case #%d:
",t );
        work();
    }
    return 0;
}