bzoj4556: [Tjoi2016&Heoi2016]字符串(二分答案+sam+线段树合并)

传送门
题意:给一个字符串 S S S
m m m次询问,每次给四个参数 a , b , c , d a,b,c,d a,b,c,d,问 s [ a . . . b ] s[a...b] s[a...b]的所有子串和 s [ x . . . y ] s[x...y] s[x...y]的最长公共前缀是多少。


思路:先翻转字符串转化为求最长公共后缀。
设现在求 s [ a . . . b ] s[a...b] s[a...b]的所有子串和 s [ x . . . y ] s[x...y] s[x...y]的最长公共后缀是多少。
然后二分答案,设最长公共后缀为 s [ y − m i d + 1... y ] s[y-mid+1...y] s[ymid+1...y],我们在反串的 s a m sam sam倍增找到这个串对应的节点,然后如果存在 s [ a . . b ] s[a..b] s[a..b]的子串那么这个子串的右端点一定在这个倍增出的节点的 r i g h t right right集合中,于是对于每个节点用线段树合并处理出它对应的 r i g h t right right集合最后查询即可。
代码:

#include<bits/stdc++.h>
#define ri register int
using namespace std;
const int N=3e5+5,M=5e6+5;
int rt[N],pos[N],st[N][20],n,m;
namespace SGT{
	#define lc (son[p][0])
	#define rc (son[p][1])
	#define mid (l+r>>1)
	int son[M][2],tot=0;
	bool exi[M];
	inline void build(int&p,int l,int r,int k){
		if(!p)p=++tot;
		exi[p]=1;
		if(l==r)return;
		k<=mid?build(lc,l,mid,k):build(rc,mid+1,r,k);
	}
	inline int merge(int x,int y,int l,int r){
		if(!x||!y)return x+y;
		int p=++tot;
		exi[p]=exi[x]|exi[y];
		if(l==r)return p;
		lc=merge(son[x][0],son[y][0],l,mid);
		rc=merge(son[x][1],son[y][1],mid+1,r);
		return p;
	}
	inline bool query(int p,int l,int r,int ql,int qr){
		if(!exi[p])return 0;
		if(ql<=l&&r<=qr)return exi[p];
		if(qr<=mid)return query(lc,l,mid,ql,qr);
		if(ql>mid)return query(rc,mid+1,r,ql,qr);
		return query(lc,l,mid,ql,qr)|query(rc,mid+1,r,ql,qr);
	}
	#undef lc
	#undef rc
	#undef mid
}
namespace sam{
	int link[N],len[N],son[N][26],tot=1,last=1;
	inline void expand(int x,int id){
		int p=last,np=++tot;
		pos[id]=last=np,len[np]=len[p]+1,SGT::build(rt[np],1,n,id);
		while(p&&!son[p][x])son[p][x]=np,p=link[p];
		if(!p){link[np]=1;return;}
		int q=son[p][x],nq;
		if(len[q]==len[p]+1){link[np]=q;return;}
		len[nq=++tot]=len[p]+1,link[nq]=link[q],memcpy(son[nq],son[q],sizeof(son[q]));
		link[q]=link[np]=nq;
		while(p&&son[p][x]==q)son[p][x]=nq,p=link[p];
	}
	inline void init(){
		static int cnt[N],rk[N];
		for(ri i=1;i<=tot;++i)st[i][0]=link[i];
		for(ri j=1;j<20;++j)for(ri i=1;i<=tot;++i)st[i][j]=st[st[i][j-1]][j-1];
		for(ri i=1;i<=tot;++i)++cnt[len[i]];
		for(ri i=1;i<=tot;++i)cnt[i]+=cnt[i-1];
		for(ri i=tot;i;--i)rk[cnt[len[i]]--]=i;
		for(ri i=tot;i^1;--i)rt[link[rk[i]]]=SGT::merge(rt[link[rk[i]]],rt[rk[i]],1,n);
	}
}
inline bool check(int a,int b,int lim,int r){
	int p=pos[r];
	for(ri i=19;~i;--i)if(sam::len[st[p][i]]>=lim)p=st[p][i];
	return SGT::query(rt[p],1,n,a,b);
}
char s[N];
inline int read(){
	int ans=0;
	char ch=getchar();
	while(!isdigit(ch))ch=getchar();
	while(isdigit(ch))ans=(ans<<3)+(ans<<1)+(ch^48),ch=getchar();
	return ans;
}
int main(){
	n=read(),m=read();
	scanf("%s",s+1),reverse(s+1,s+n+1);
	for(ri i=1;i<=n;++i)sam::expand(s[i]-'a',i);
	sam::init();
	for(ri i=1,a,b,x,y,l,r,mid,ans;i<=m;++i){
		b=n-read()+1,a=n-read()+1,y=n-read()+1,x=n-read()+1;
		l=1,r=min(b-a+1,y-x+1),ans=0;
		while(l<=r)if(mid=l+r>>1,check(a+mid-1,b,mid,y))l=mid+1,ans=mid;else r=mid-1;
		cout<<ans<<'\n';
	}
	return 0;
}
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值