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+10;
const int D = 21; 
const int maxnode = 1800010; 

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

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

struct SAM
{
    int sz,fa[maxnode],mx,ch[maxnode][26],l[maxnode];
    int siz[maxnode],b[maxnode],c[N],fat[maxnode][D];
    SAM() {
        sz=1;
        memset(l,0,sizeof(l));
        memset(siz,0,sizeof(siz));
    }
    int add(int c,int p) {
        int np=++sz; l[np]=l[p]+1; 
        mx=max(mx,l[np]); siz[np]=1;
        for(;p&&!ch[p][c];p=fa[p]) ch[p][c]=np;
        if(!p) fa[np]=1;
        else {
            int q=ch[p][c];
            if(l[q]==l[p]+1) fa[np]=q;
            else {
                int nq=++sz; l[nq]=l[p]+1;
                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,1,sz) c[l[i]]++;
        FOR(i,1,mx) c[i]+=c[i-1];
        FOR(i,1,sz) b[c[l[i]]--]=i;
        rep(i,sz,1) siz[fa[b[i]]]+=siz[b[i]];
        FOR(j,1,sz) {
            int i=b[j];                                //按照拓扑序递推fat 
            fat[i][0]=fa[i];
            FOR(j,1,D-1)
                fat[i][j]=fat[fat[i][j-1]][j-1];
        }
    }
    int solve(int u,int len) {
        rep(i,D-1,0)
            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][26];
    queue<Node> q;
    Trie() {
        sz=0;
        memset(ch,0,sizeof(ch));
    }
    void insert(char* s) {
        int u=0;
        for(int i=0;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=0;
        q.push((Node){u,1});
        while(!q.empty()) {
            Node qr=q.front(); q.pop();
            int u=qr.id,p=qr.np;
            FOR(c,0,25) 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,1,n) {
        scanf("%s",s);
        len[i]=len[i-1]+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-1],y+=len[p-1];
        printf("%d
",sam.solve(pos2[pos[y]],y-x+1));
    }
    return 0;
}