bzoj 3413: 匹配

Description

Input

第一行包含一个整数n(≤100000)。

第二行是长度为n的由0到9组成的字符串。

第三行是一个整数m。

接下来m≤5·10^4行，第i行是一个由0到9组成的字符串s，保证单行字符串长度小于等于10^5，所有字符串长度和小于等于3·10^6

Output

输出m行，第i行表示第si和S匹配所比较的次数。

建出后缀树，将询问按第一次匹配到的位置从小到大排序，离线处理，用树状数组维护后缀树的dfs序

从左到右加入后缀，加入时将后缀树上对应点+1，处理第一次匹配位置在这个后缀的询问

对每个询问串，在后缀树上匹配一次，查询经过的每个点的子树和

#include<bits/stdc++.h>
typedef long long i64;
const int N=2e5+;
int n,m;
char s1[N];
int nx[N][],ch[N][],l[N],fa[N],pv=,ptr=,lr[N][],in[N],q[N],ql,qr,id[N][],idp=;
int bit[N],bs[N],ws[N];
void inc(int w){
    for(++bs[w];w<=idp;w+=w&-w)++bit[w];
}
bool _=;
int sum(int w){
    if(_)return bs[w];
    int s=;
    for(;w;w-=w&-w)s+=bit[w];
    return s;
}
void ins(int pos){
    int x=s1[pos]-='';
    int p=pv,np=++ptr;
    l[np]=l[p]+;
    while(p&&!nx[p][x])nx[p][x]=np,p=fa[p];
    if(!p)fa[np]=;
    else{
        int q=nx[p][x];
        if(l[q]==l[p]+)fa[np]=q;
        else{
            int nq=++ptr;
            lr[nq][]=n+;
            l[nq]=l[p]+;
            memcpy(nx[nq],nx[q],sizeof(nx[]));
            fa[nq]=fa[q];
            fa[q]=fa[np]=nq;
            while(p&&nx[p][x]==q)nx[p][x]=nq,p=fa[p];
        }
    }
    ws[pos]=np;
    lr[np][]=n;
    pv=np;
}
void dfs(int w){
    id[w][]=++idp;
    for(int i=;i<;++i)if(ch[w][i])dfs(ch[w][i]);
    id[w][]=idp;
}
void mins(int&a,int b){if(a>b)a=b;}
i64 as[];
struct Q{
    char*s;
    int len,e,ID;
    void match(){
        int w=,p=;
        for(int i=;i<len&&w;++i){
            int x=s[i]-'';
            if(p==lr[w][])p=lr[w=ch[w][x]][];
            else if(s1[++p]!=x)w=;
        }
        e=w?p-len+:n+;
    }
    void cal(){
        int w=,p=;
        for(int i=;i<len&&w;++i){
            int x=s[i]-'';
            if(p==lr[w][])p=lr[w=ch[w][x]][];
            else if(s1[++p]!=x)w=;
            as[ID]+=sum(id[w][])-sum(id[w][]-);
        }
        as[ID]+=e>n?n:e-;
    }
    bool operator<(Q x)const{return e<x.e;}
}qs[];
char s2[],*sp2=s2;
int main(){
    scanf("%d%s",&n,s1+);
    for(int i=n;i;--i)ins(i);
    for(int i=;i<=ptr;++i)++in[fa[i]];
    for(int i=;i<=ptr;++i)if(!in[i])q[++qr]=i;
    in[]=-;
    while(ql!=qr){
        int w=q[++ql],f=fa[w];
        if(!--in[f])q[++qr]=f;
        lr[w][]=lr[w][]-l[w]+l[f]+;
        mins(lr[f][],lr[w][]-);
        ch[f][s1[lr[w][]]]=w;
    }
    dfs();
    scanf("%d",&m);
    for(int i=;i<m;++i){
        scanf("%s",sp2);
        qs[i].s=sp2;
        qs[i].len=strlen(sp2);
        qs[i].match();
        qs[i].ID=i;
        sp2+=qs[i].len+;
    }
    std::sort(qs,qs+m);
    int qp=;
    for(int i=;i<=n;++i){
        inc(id[ws[i]][]);
        for(;qp<m&&qs[qp].e==i;++qp)qs[qp].cal();
    }
    _=;
    for(int i=;i<=idp;++i)bs[i]+=bs[i-];
    for(;qp<m;++qp)qs[qp].cal();
    for(int i=;i<m;++i)printf("%lld\n",as[i]);
    return ;
}