【BZOJ2780】Sevenk Love Oimaster【广义后缀自动机】

题意

给出你n个字符串和q个查询，每个查询给出一个字符串s，对于每个查询你都要输出这个字符串s在上面多少个字符串中出现过。

分析

广义后缀自动机的裸题。建好SAM以后再跑一遍得到每个状态的ocu和las。然后对于每个查询的字符串，跑到那个状态然后输出那个状态的ocu就可以了。

 #include <cstdio>
 #include <cstring>
 #include <algorithm>
 #include <iostream>

 using namespace std;
 const int maxn=;
 struct state{
     int len,link,ocu,las;
     int next[];
 }st[maxn];
 int N,Q,n;
 int last,cur,sz;
 char s[maxn];
 string S[+];
 void init(){
     sz=;
     cur=last=;
     st[].link=-;
     st[].len=;
 }
 void build_sam(int c){
     cur=sz++;
     st[cur].len=st[last].len+;
     int p;
     for(p=last;p!=-&&st[p].next[c]==;p=st[p].link){
             st[p].next[c]=cur;
     }
     if(p==-)
         st[cur].link=;
     else{
         int q=st[p].next[c];
         if(st[q].len==st[p].len+)
             st[cur].link=q;
         else{
             int clone=sz++;
             st[clone].len=st[p].len+;
             st[clone].link=st[q].link;
             for(int i=;i<;i++)
                 st[clone].next[i]=st[q].next[i];
             for(;p!=-&&st[p].next[c]==q;p=st[p].link)
                 st[p].next[c]=clone;
             st[cur].link=st[q].link=clone;
         }
     }
     last=cur;
 }

 int main(){
     scanf("%d%d",&N,&Q);
     init();
     for(int i=;i<=N;i++){
         scanf("%s",s);
         S[i]=(string)s;
         n=strlen(s);
         for(int j=;j<n;j++){
             build_sam(s[j]-'a');
         }
         last=;
     }

     for(int i=;i<=N;i++){
         int u=;
         for(int j=;j<S[i].length();j++){
             u=st[u].next[S[i][j]-'a'];
             int p=u;
             while(p!=-&&st[p].las!=i){
                 st[p].ocu++;
                 st[p].las=i;
                 p=st[p].link;
             }
         }
     }

     for(int q=;q<=Q;q++){
         scanf("%s",s);
         n=strlen(s);
         int u=,flag=;
         for(int i=;i<n;i++){
             if(st[u].next[s[i]-'a']==){
                 flag=;
                 break;
             }
             u=st[u].next[s[i]-'a'];
         }
         if(!flag){
             printf("0\n");
         }else
         printf("%d\n",st[u].ocu);
     }

 return ;
 }