BZOJ5084: hashit

后缀自动机

题目传送门

询问答案的话直接算就好了。

撤销的话记录一个指针数组$a$指向每次插入后改变的节点，另开一个$b$存修改前的值。当撤销的时候只需要$^*a[i]=b[i]$即可。

其实就是暴力（但是跑得比正解快）。

代码：

#include<cctype>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define N 200005
using namespace std;
typedef long long LL;
int lt,nd,t,l[N],nxt[N],to[N][26],*a[N],b[N],s[N];
char q[N]; LL ans; bool f[N];
void writec(LL x){ if (x>9) writec(x/10); putchar(x%10+48); }
void _write(LL x){ writec(x),putchar('\n'); }
void add(int &x){ a[++t]=&x,b[t]=x,s[nd]++; }
void nsrt(int c){
    int x=++nd,p=lt,q; l[lt=x]=l[p]+1,s[nd]=0;
    memset(to[x],0,sizeof(to[x])),f[nd]=0;
    while (p&&!to[p][c]) add(to[p][c]),to[p][c]=x,p=nxt[p];
    if (!p) nxt[x]=1;
    else
        if (l[q=to[p][c]]==l[p]+1) nxt[x]=q;
        else{
            int y=++nd; l[y]=l[p]+1,f[nd]=1,s[nd]=0;
            memcpy(to[y],to[q],sizeof(to[y]));
            nxt[y]=nxt[q],add(nxt[q]),nxt[q]=nxt[x]=y;
            while (to[p][c]==q) add(to[p][c]),to[p][c]=y,p=nxt[p];
        }
    ans+=l[x]-l[nxt[x]];
}
int main(){
    scanf("%s",q+1),nd=lt=1;
    for (int i=1;q[i];i++){
        if (islower(q[i])) nsrt(q[i]-'a');
        else
            for (int j=0,k=f[nd];j<=k;j++){
                for (int p=1;p<=s[nd];p++,t--) *a[t]=b[t];
                if (!f[nd]) ans-=l[nd]-l[nxt[nd]]; lt=--nd;
            }
        if (f[lt]) lt--; _write(ans);
    }
    return 0;
}