51nod 小Z的trie（Trie+广义SAM）

【题目链接】

http://www.51nod.com/contest/problem.html#!problemId=1647

【题意】

给定一个n个字符串的Trie，每次询问一个字符串在Trie上的出现次数。

【思路】

将n个字符串构造一个Trie，构造广义后缀自动机，识别Trie中的所有子串。

则问题转化为求子串[l,r]在所属字符串中的出现次数，和 这道题 类似，采用树上倍增得到|right|的方法。

　 比较恶心的是定位字符串的位置=-=。

【代码】

 #include<set>
 #include<cmath>
 #include<queue>
 #include<vector>
 #include<cstdio>
 #include<cstring>
 #include<iostream>
 #include<algorithm>
 #define trav(u,i) for(int i=front[u];i;i=e[i].nxt)
 #define FOR(a,b,c) for(int a=(b);a<=(c);a++)
 #define rep(a,b,c) for(int a=(b);a>=(c);a--)
 using namespace std;

 typedef long long ll;
 const int N = 1e6+;
 const int D = ;
 const int maxnode = ; 

 int pos[N],pos2[N],cur;

 ll read() {
     char c=getchar();
     ll f=,x=;
     while(!isdigit(c)) {
         if(c=='-') f=-; c=getchar();
     }
     while(isdigit(c))
         x=x*+c-'',c=getchar();
     return x*f;
 }

 struct SAM
 {
     int sz,fa[maxnode],mx,ch[maxnode][],l[maxnode];
     int siz[maxnode],b[maxnode],c[N],fat[maxnode][D];
     SAM() {
         sz=;
         memset(l,,sizeof(l));
         memset(siz,,sizeof(siz));
     }
     int add(int c,int p) {
         int np=++sz; l[np]=l[p]+;
         mx=max(mx,l[np]); siz[np]=;
         for(;p&&!ch[p][c];p=fa[p]) ch[p][c]=np;
         if(!p) fa[np]=;
         else {
             int q=ch[p][c];
             if(l[q]==l[p]+) fa[np]=q;
             else {
                 int nq=++sz; l[nq]=l[p]+;
                 memcpy(ch[nq],ch[q],sizeof(ch[q]));
                 fa[nq]=fa[q];
                 fa[np]=fa[q]=nq;
                 for(;q==ch[p][c];p=fa[p]) ch[p][c]=nq;
             }
         }
         return np;
     }
     void get_pre() {
         FOR(i,,sz) c[l[i]]++;
         FOR(i,,mx) c[i]+=c[i-];
         FOR(i,,sz) b[c[l[i]]--]=i;
         rep(i,sz,) siz[fa[b[i]]]+=siz[b[i]];
         FOR(j,,sz) {
             int i=b[j];                                //按照拓扑序递推fat
             fat[i][]=fa[i];
             FOR(j,,D-)
                 fat[i][j]=fat[fat[i][j-]][j-];
         }
     }
     int solve(int u,int len) {
         rep(i,D-,)
             if(l[fat[u][i]]>=len) u=fat[u][i];
         return siz[u];
     }
 } sam;

 struct Node {
     int id,np;
 };
 struct Trie
 {
     int sz,ch[N][];
     queue<Node> q;
     Trie() {
         sz=;
         memset(ch,,sizeof(ch));
     }
     void insert(char* s) {
         int u=;
         for(int i=;s[i];i++) {
             int c=s[i]-'a';
             if(!ch[u][c]) ch[u][c]=++sz;
             u=ch[u][c]; pos[++cur]=u;
         }
     }
     void bfs() {
         int u=;
         q.push((Node){u,});
         while(!q.empty()) {
             Node qr=q.front(); q.pop();
             int u=qr.id,p=qr.np;
             FOR(c,,) if(ch[u][c]) {
                 int v=ch[u][c];
                 int np=sam.add(c,p);
                 pos2[v]=np;
                 q.push((Node){v,np});
             }
         }
     }
 } trie;

 int n,m,len[N];
 char s[N];

 int main()
 {
     n=read();
     FOR(i,,n) {
         scanf("%s",s);
         len[i]=len[i-]+strlen(s);
         trie.insert(s);
     }
     trie.bfs();
     sam.get_pre();
     scanf("%d",&m);
     int p,x,y;
     while(m--) {
         p=read(),x=read(),y=read();
         x+=len[p-],y+=len[p-];
         printf("%d\n",sam.solve(pos2[pos[y]],y-x+));
     }
     return ;
 }