【校内模拟】子串(SAM)(LCT)(主席树)

34 篇文章 0 订阅
29 篇文章 0 订阅

题解:

考虑怎么算答案。

首先本质不同子串个数大概就是几种做法,后缀数组,后缀自动机,暴力

后缀数组用不了了,强制在线令人头疼。

然后后缀自动机。。。众所周知,并没有区间后缀自动机这种扯淡的数据结构。

换一个思路,我们并不直接统计,而是对于前面每一个重复出现的字符串,打上-1标记。

仔细想一想即可得到本题正解。

怎么得到重复字符串,显然是SAM跳fail链,每个点维护right集合最大值,更新之前先把前面的-1打上标记,对于每个右端点可持久化一下,主席树标记永久化维护区间加区间求和。

发现right集合最大值相同的更新起来的是连续的,直接用LCT维护最大值相同的链,那么新加一个点直接access,在access的同时在主席树里面修改即可。

由Finger Search 理论,复杂度只有一个 O ( n log ⁡ n ) O(n\log n) O(nlogn)

最后实现注意一下字符串边界,LCT稍微考虑一下可以不用写cut。

其实就是在断掉父亲和它之间的实边之后,将它Splay,然后直接改父亲指针,也就是子树换父亲(我真TM机智)。


代码:

#include<bits/stdc++.h>
#define ll long long
#define re register
#define cs const

namespace IO{
	cs int Rlen=1<<22|1;
	char buf[Rlen],*p1,*p2;
	inline char gc(){return (p1==p2)&&(p2=(p1=buf)+fread(buf,1,Rlen,stdin),p1==p2)?EOF:*p1++;}
	inline char peek(){return (p1==p2)&&(p2=(p1=buf)+fread(buf,1,Rlen,stdin),p1==p2)?EOF:*p1;}
	inline char ga(){while(!isalpha(peek()))++p1;return *p1++;}
	template<typename T>
	inline T get(){
		char c;T num;
		while(!isdigit(c=gc()));num=c^48;
		while(isdigit(c=gc()))num=(num+(num<<2)<<1)+(c^48);
		return num;
	}
	inline int gi(){return get<int>();}
	inline ll gl(){return get<ll>();}
}
using namespace IO;

using std::cerr;
using std::cout;

cs int N=1e5+7;

int M;
namespace PST{
	cs int N=::N*80;
	int lc[N],rc[N],tag[N],tot;ll sum[N];
	inline int _copy(int u){++tot;lc[tot]=lc[u],rc[tot]=rc[u];tag[tot]=tag[u],sum[tot]=sum[u];return tot;}
	inline void ins(int &u,int l,int r,int ql,int qr,int v){
		u=_copy(u);sum[u]+=(ll)(std::min(qr,r)-std::max(ql,l)+1)*v;
		if(ql<=l&&r<=qr){tag[u]+=v;return ;}
		int mid=l+r>>1;
		if(ql<=mid)ins(lc[u],l,mid,ql,qr,v);
		if(mid<qr)ins(rc[u],mid+1,r,ql,qr,v);
	}
	inline ll query(int u,int l,int r,int ql,int qr){
		if(!u)return 0;if(ql<=l&&r<=qr)return sum[u];
		int mid=l+r>>1;
		ll tp=(ll)(std::min(qr,r)-std::max(ql,l)+1)*tag[u];
		if(qr<=mid)return tp+query(lc[u],l,mid,ql,qr);
		if(mid<ql)return tp+query(rc[u],mid+1,r,ql,qr);
		return tp+query(lc[u],l,mid,ql,qr)+query(rc[u],mid+1,r,ql,qr);
	}
}

int rt[N];

namespace LCT{
	cs int N=::N<<1|1;
	int fa[N],son[N][2],las[N],tag[N],len[N];
	inline void pushtag(int u,int v){tag[u]=las[u]=v;}
	inline void pushdown(int u){
		if(tag[u]){
			pushtag(son[u][0],tag[u]);
			pushtag(son[u][1],tag[u]);
			tag[u]=0;
		}
	}
	inline bool isrt(int u){return son[fa[u]][0]!=u&&son[fa[u]][1]!=u;}
	inline bool which(int u){return son[fa[u]][1]==u;}
	inline void Rotate(int u){
		int p=fa[u],pp=fa[p],d=which(u);
		if(!isrt(p))son[pp][which(p)]=u;
		fa[u]=pp,fa[p]=u,fa[son[u][!d]]=p;
		son[p][d]=son[u][!d],son[u][!d]=p;
	}
	inline void Splay(int u){
		static int q[N],qn;q[qn=1]=u;
		for(int re p=u;!isrt(p);p=fa[p])q[++qn]=fa[p];
		while(qn)pushdown(q[qn--]);
		for(int re p=fa[u];!isrt(u);Rotate(u),p=fa[u])
		if(!isrt(p))Rotate(which(u)==which(p)?p:u);
	}
	inline void access(int u,int id){
		int ch=0;for(;u;u=fa[ch=u]){
			Splay(u);if(las[u]){
				PST::ins(rt[id],1,M,las[u]-len[u]+1,las[u]-len[fa[u]],1);
			}
			son[u][1]=ch;
		}
		pushtag(ch,id);
	}
	inline int get(int u){Splay(u);return las[u];}
	inline void link(int u,int f){Splay(u);fa[u]=f;}
}

int nowl;
namespace SAM{
	cs int N=::N<<1|1;
	int son[N][26],fa[N],len[N];
	int now=1,last=1;
	inline void init_LCT(int u){
		LCT::len[u]=len[u];
	}
	inline void push_back(int c){
		int id=++nowl,cur=++now,p=last;rt[id]=rt[id-1],last=cur;
		len[cur]=len[p]+1;init_LCT(cur);
		for(;p&&!son[p][c];p=fa[p])son[p][c]=cur;
		if(!p)
			LCT::link(cur,fa[cur]=1),
			LCT::access(cur,id);
		else if(len[son[p][c]]==len[p]+1)
			LCT::link(cur,fa[cur]=son[p][c]),
			LCT::access(cur,id);
		else {
			int nq=++now,q=son[p][c];
			memcpy(son[nq],son[q],sizeof son[q]);
			len[nq]=len[p]+1;init_LCT(nq);
			LCT::las[nq]=LCT::get(q);
			LCT::link(nq,fa[nq]=fa[q]);
			LCT::link(cur,fa[cur]=nq);
			LCT::access(cur,id);
			LCT::link(q,fa[q]=nq);
			for(;p&&son[p][c]==q;p=fa[p])son[p][c]=nq;
		}
	}
}

ll ans;
int n,m,op;
char s[N];

inline void ins(){
	int c=ga()-'a';
	if(op)c=(c+ans)%26;
	SAM::push_back(c);
}

inline void query(){
	int l=gi(),r=gi();
	if(op)l=(l-1+ans)%nowl+1,r=(r-1+ans)%nowl+1;
	int Len=r-l+1;
	ans=(Len+1ll)*(Len)/2-PST::query(rt[r],1,M,l,r);
	cout<<ans<<"\n";
}

#ifdef zxyoi
#undef zxyoi
#endif

signed main(){
#ifdef zxyoi
	freopen("substring.in","r",stdin);
#else
#ifndef ONLINE_JUDGE
	freopen("substring.in","r",stdin);freopen("substring.out","w",stdout);
#endif
#endif
	scanf("%d%s%d",&op,s+1,&m);
	n=strlen(s+1);M=n+m;
	for(int re i=1;i<=n;++i)SAM::push_back(s[i]-'a');
	while(m--){
		switch(gi()){
			case 1:ins();break;
			case 2:query();break;
		}
	}
	return 0;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值