HDU6054 String and String 广义后缀自动机+树状数组套线段树

存一下代码。
广义SAM是正版的。

Code

#pragma comment(paintr, "/STACK:102400000,102400000") 
#include<bits/stdc++.h>
using namespace std;
#define LL long long
#define pa pair<int,int>
const int Maxn=400010,N=390;
const int inf=2147483647;
const double pi=acos(-1.0);
LL read()
{
	LL 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*10ll+ch-'0',ch=getchar();
	return x*f;
}
int c[Maxn*N],lc[Maxn*N],rc[Maxn*N],root[Maxn],cnt;
void modify(int&u,int l,int r,int p,int v)
{
	if(!u)u=++cnt;
	if(l==r)
	{
		c[u]=v;
		return;
	}
	int mid=l+r>>1;
	if(p<=mid)modify(lc[u],l,mid,p,v);
	else modify(rc[u],mid+1,r,p,v);
	c[u]=c[lc[u]]+c[rc[u]];
}
int query(int x,int l,int r,int ql,int qr)
{
	if(!x)return 0;
	if(l==ql&&r==qr)return c[x];
	int mid=l+r>>1;
	if(qr<=mid)return query(lc[x],l,mid,ql,qr);
	if(ql>mid)return query(rc[x],mid+1,r,ql,qr);
	return query(lc[x],l,mid,ql,mid)+query(rc[x],mid+1,r,mid+1,qr);
}
char S[Maxn],T[Maxn];int ls,lt,f[Maxn],Spos[Maxn],Tpos[Maxn];
int last,tot,son[Maxn][26],par[Maxn],mx[Maxn];
int extend(int ch,int last)
{
    if(son[last][ch])
	{
        int p=last,x=son[p][ch];
        if(mx[p]+1==mx[x])return x;
        else
		{
            int y=++tot;mx[y]=mx[p]+1;
            for(int i=0;i<26;++i)son[y][i]=son[x][i];
            while(p&&son[p][ch]==x)son[p][ch]=y,p=par[p];
            par[y]=par[x],par[x]=y;
            return y;
        }
    }
    int z=++tot,p=last;mx[z]=mx[last]+1;
    while(p&&!son[p][ch])son[p][ch]=z,p=par[p];
    if(!p)par[z]=1;
    else
	{
        int x=son[p][ch];
        if(mx[p]+1==mx[x])par[z]=x;
        else
		{
            int y=++tot;mx[y]=mx[p]+1;
            for(int i=0;i<26;++i)son[y][i]=son[x][i];
            while(p&&son[p][ch]==x)son[p][ch]=y,p=par[p];
            par[y]=par[x],par[z]=par[x]=y;
        }
    }
    return z;
}
//树状数组层:dfs序 线段树层:|S| 
void Modify(int x,int y,int v)
{
	for(;x<=tot;x+=(x&(-x)))modify(root[x],1,ls,y,v);
}
int Query(int x1,int x2,int y1,int y2)
{
	if(x1>x2||y1>y2)return 0;
	int t=0;
	for(;x2;x2-=(x2&(-x2)))t+=query(root[x2],1,ls,y1,y2);
	--x1;
	for(;x1;x1-=(x1&(-x1)))t-=query(root[x1],1,ls,y1,y2);
	return t;
}
vector<int>G[Maxn];
int L[Maxn],R[Maxn],dfn,fa[Maxn][20],dep[Maxn];
void dfs(int x)
{
	L[x]=++dfn;
	for(int i=1;(1<<i)<=dep[x];i++)fa[x][i]=fa[fa[x][i-1]][i-1];
	for(int i=0;i<G[x].size();i++)
	{
		int y=G[x][i];
		dep[y]=dep[x]+1;
		fa[y][0]=x;
		dfs(y);
	}
	R[x]=dfn;
}
int jump(int a,int b)
{
	int t=Tpos[b];
	for(int i=19;i>=0;i--)
	if((1<<i)<=dep[t]&&mx[fa[t][i]]>=b-a+1)t=fa[t][i];
	return t;
}
int main()
{
	int Case=read();
	while(Case--)
	{
		last=tot=1;dfn=cnt=0;
		scanf("%s%s",S+1,T+1);
		ls=strlen(S+1),lt=strlen(T+1);
		for(int i=1;i<=ls;i++)Spos[i]=last=extend(S[i]-'a',last);
		last=1;
		for(int i=1;i<=lt;i++)Tpos[i]=last=extend(T[i]-'a',last);
		for(int i=1;i<=tot;i++)G[i].clear();
		for(int i=2;i<=tot;i++)G[par[i]].push_back(i);
		dep[1]=0;dfs(1);
		for(int i=1;i<=ls;i++)Modify(L[Spos[i]],i,read());
		int ans=0,q=read();
		while(q--)
		{
			int op=read(),a=read(),b=read(),c,d;
			a^=ans,b^=ans;
			if(op==1)
			{
				a++;
				Modify(L[Spos[a]],a,b);
			}
			else
			{
				c=read(),d=read();
				c^=ans,d^=ans;a++,b++,c++,d++;
				int p=jump(a,b);
				ans=Query(L[p],R[p],c+b-a,d);
				printf("%d\n",ans);
			}
		}
		for(int i=1;i<=tot;i++)
		{
			root[i]=mx[i]=par[i]=0;
			for(int j=0;j<26;j++)son[i][j]=0;
		}
		for(int i=1;i<=cnt;i++)c[i]=lc[i]=rc[i]=0;
	}
}