51nod 1647 小Z的trie 广义后缀自动机+倍增

题意

NOIP编号为ZJ-267的小Z在NOIP中AK啦！
小Z打算去冲击省选，于是开始学习trie。
有一天，他得到了N个字符串。
他先建立一个根节点，对于每一个字符串，他都从根节点开始一点点插入。
小Z不满足于此。他的大脑里盘旋着M个问题：
如果给定一个二元组（s,t）（s，t都是trie中的节点且s是t的祖先），
存在多少个二元组（x,y）（x，y都是trie中的节点且x是y的祖先），
满足s~t路径上的字符串和x~y路径上的字符串完全一样？
注意s可以等于t，x也可以等于y。

这里为了方便，读入仅仅是N个字符串，请自己建立出trie。
同时每一组询问的格式为（pi，xi，yi），
第pi个串的第xi个字符在trie中的位置即为s，
第pi个串的第yi个字符在trie中的位置即为t。

第一行为一个整数N。（N<=100000）
接下来N行，每行一个字符串。（保证字符串总长<=1000000)
接下来一行一个整数M。（M<=100000）
接下来M行，每行三个数pi，xi，yi意义如上。
（1<=pi<=N，1<=xi<=yi<=|Spi|）

分析

一开始想到了一个后缀数组+二分+主席树的做法，看了题解发现可以用后缀自动机来做。
我们可以在trie上建sam，记录每个位置在sam上对应的节点。对于一次询问，我们可以从深度较大的位置对应的节点开始，在parents树上往上跳，直到一个节点满足其区间包含询问字符串的长度。跳的过程可以用倍增来实现。

代码

#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<algorithm>
#include<vector>
using namespace std;

const int N=100005;

int n,m,ch[N*20][26],size[N*20],fa[N*20],bz[N*20][21],mx[N*20],b[N*20],c[N*20],trie[N*10][26],cnt,tot,ls[N*10];
vector<int> vec[N];
char str[N*10];

int read()
{
    int x=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while (ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}

void ins(int &last,int x)
{
    if (ch[last][x])
    {
        int p=last,np=ch[p][x];
        if (mx[np]==mx[p]+1) last=np;
        else
        {
            int q=++cnt;mx[q]=mx[p]+1;
            memcpy(ch[q],ch[np],sizeof(ch[q]));
            fa[q]=fa[np];
            fa[np]=last=q;
            for (;ch[p][x]==np;p=fa[p]) ch[p][x]=q;
        }
        size[last]++;
        return;
    }
    int p,q,np,nq;
    p=last;last=np=++cnt;mx[np]=mx[p]+1;size[np]=1;
    for (;p&&!ch[p][x];p=fa[p]) ch[p][x]=np;
    if (!p) fa[np]=1;
    else
    {
        q=ch[p][x];
        if (mx[q]==mx[p]+1) fa[np]=q;
        else
        {
            nq=++cnt;mx[nq]=mx[p]+1;
            fa[nq]=fa[q];
            fa[q]=fa[np]=nq;
            memcpy(ch[nq],ch[q],sizeof(ch[nq]));
            for (;ch[p][x]==q;p=fa[p]) ch[p][x]=nq;
        }
    }
}

void prework()
{
    for (int i=1;i<=cnt;i++) b[mx[i]]++;
    for (int i=1;i<=cnt;i++) b[i]+=b[i-1];
    for (int i=cnt;i>=1;i--) c[b[mx[i]]--]=i;
    for (int i=cnt;i>=1;i--) size[fa[c[i]]]+=size[c[i]];
    for (int i=1;i<=cnt;i++)
    {
        int x=c[i];
        bz[x][0]=fa[x];
        for (int j=1;j<=20;j++) bz[x][j]=bz[bz[x][j-1]][j-1];
    }
}

int main()
{
    n=read();ls[0]=cnt=1;
    for (int i=1;i<=n;i++)
    {
        scanf("%s",str);
        int len=strlen(str);
        int now=0;
        for (int j=0;j<len;j++)
        {
            int x=str[j]-'a';
            if (trie[now][x]) now=trie[now][x];
            else trie[now][x]=++tot,ls[tot]=ls[now],ins(ls[tot],x),now=tot;
            vec[i].push_back(ls[now]);
        }
    }
    prework();
    m=read();
    while (m--)
    {
        int p=read(),x=read(),y=read();
        int t=vec[p][y-1],len=y-x+1;
        for (int i=20;i>=0;i--) if (mx[bz[t][i]]>=len) t=bz[t][i];
        printf("%d\n",size[t]);
    }
    return 0;
}