LOJ 6070基因 回文树 分块

提交链接

题解

分为$\sqrt{n}$ 个块，每个块预处理出起点到 n 的回文树，需要保存的信息有区间的回文串种类个数、区间的某种结点出现的最早位置以及区间的最长回文前缀对应的结点
后两个都是为了向前插入准备的
这是因为，如果向前插入的过程中，产生的一个新的结点，我们需要看这个结点在区间内是否出现过，
此外，向前插入，需要从最长回文前缀开始

这道题需要对回文树有深入的理解，并且要用到翁文涛在论文里讲的另一种快速插入的方式 $quick$,所以最好对论文有深入理解才能解决本题

代码

#include<bits/stdc++.h>
#define N 100010
#define INF 0x3f3f3f3f
#define eps 1e-6
#define pi 3.141592653589793
#define mod 1000000007
#define P 1000000007
#define LL long long
#define pb push_back
#define fi first
#define se second
#define cl clear
#define si size
#define lb lower_bound
#define ub upper_bound
#define bug(x) cerr<<#x<<"      :   "<<x<<endl
#define mem(x,y) memset(x,0,sizeof(int)*(y+3))
#define sc(x) scanf("%d",&x)
#define scc(x,y) scanf("%d%d",&x,&y)
#define sccc(x,y,z) scanf("%d%d%d",&x,&y,&z)
using namespace std;
typedef  pair<int,int> pp;
char s[N];
int p[320][N],ans[320][N],pos[320][N];
int vis[N],fail[N],len[N],a[N],nxt[N][26],qic[N][26],last,head,cnt,T;

void init(){
    fail[0]=fail[1]=1,len[1]=-1;
    for(int i=0;i<26;i++) qic[0][i]=1;
    cnt=2;
}
inline int newnode(int l){
    len[cnt]=l;
    return cnt++;
}
void exback(int l,int i){
    int c=a[i];
    if (i-len[last]-1<l||a[i-len[last]-1]!=c) last=qic[last][c];
    if(!nxt[last][c]){
        int now=newnode(len[last]+2);
        int k=fail[last];
        if (a[i-len[k]-1]!=c) k=qic[k][c]; k=nxt[k][c];
        fail[now]=k;
        nxt[last][c]=now;
        memcpy(qic[now],qic[k],sizeof qic[k]);
        qic[now][a[i-len[k]]]=k;
    }
    last=nxt[last][c];
    if (len[last]==i-l+1) head=last;
}
void exfront(int i,int r){
    int c=a[i];
    if (i+len[head]+1>r||a[i+len[head]+1]!=c) head=qic[head][c];
    if(!nxt[head][c]){
        int now=newnode(len[head]+2);
        int k=fail[head];
        if (a[i+len[k]+1]!=c) k=qic[k][c]; k=nxt[k][c];
        fail[now]=k;
        nxt[head][c]=now;
        memcpy(qic[now],qic[k],sizeof qic[k]);
        qic[now][a[i+len[k]]]=k;
    }
    head=nxt[head][c];
    if (len[head]==r-i+1) last=head;
}

int main(){
    int tpye,n,m,B;
    sccc(tpye,n,m); scanf("%s",s+1); B=sqrt(n);
    for(int i=1;i<=n;i++) a[i]=s[i]-'a';
    memset(pos,INF,sizeof pos);
    init();
    for(int L=1,i=1;L<=n;L+=B,i++){
        last=head=0; T++;
        for(int j=L;j<=n;j++){
            exback(L,j);
            ans[i][j]=ans[i][j-1];
            p[i][j]=head;
            if(vis[last]<T)
                vis[last]=T,pos[i][last]=j,ans[i][j]++;
        }
    }
    int Ans=0;
    while(m--){
        int L,R; scc(L,R);
        if (tpye) L^=Ans,R^=Ans; Ans=0;
        if ( (L-1)/B==(R-1)/B ){
            last=0; T++;
            for(int j=L;j<=R;j++){
                exback(L,j);
                if (vis[last]<T) Ans++,vis[last]=T;
            }
        }else{
            int i=(L-1)/B+1;Ans=ans[i+1][R];
            head=p[i+1][R]; T++;
            for(int j=i*B;j>=L;j--){
                exfront(j,R);
                if (vis[head]<T) Ans+=pos[i+1][head]>R,vis[head]=T;
            }
        }
        printf("%d\n",Ans);
    }
}