后缀数组+主席树--bzoj4556: [Tjoi2016&Heoi2016]字符串

传送门

可以先二分答案$mid$，判断$[a,b-mid+1]$是否有$lcp(rk[i],rk[c])\ge mid$的$i$，这个是一个区间，可以再用二分+$st$表$O(1)$查询找出来，然后用主席树判断$[a,b-mid+1]$u有没有$rk$在$[l,r]$的点

#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
#include<cmath>
#define maxn 100005
#define N 2000005
#define LL long long
using namespace std;

inline int rd(){
	int x=0,f=1;char c=getchar();
	while(c<'0' || c>'9') f=c=='-'?-1:1,c=getchar();
	while(c<='9' && c>='0') x=x*10+c-'0',c=getchar();
	return x*f;
}
inline int min(int x,int y) {return x<y?x:y;}

int n,m,a,b,c,d,q,st[maxn][20];
int tax[maxn],rk[maxn],sa[maxn],tp[maxn],h[maxn];
char s[maxn];

inline void rsort(){
	for(int i=1;i<=m;i++) tax[i]=0;
	for(int i=1;i<=n;i++) tax[rk[i]]++;
	for(int i=1;i<=m;i++) tax[i]+=tax[i-1];
	for(int i=n;i;i--) sa[tax[rk[tp[i]]]--]=tp[i];
}

inline void ssort(){
	for(int i=1;i<=n;i++) rk[i]=s[i],tp[i]=i;
	rsort();
	for(int w=1,p=0;w<=n&&p<n;w<<=1,m=p){
		p=0;
		for(int i=n-w+1;i<=n;i++) tp[++p]=i;
		for(int i=1;i<=n;i++)
			if(sa[i]>w) tp[++p]=sa[i]-w;
		rsort(); swap(rk,tp);
		rk[sa[1]]=p=1;
		for(int i=2;i<=n;i++)
			if(tp[sa[i]]==tp[sa[i-1]] && tp[sa[i]+w]==tp[sa[i-1]+w])
				rk[sa[i]]=p;
			else rk[sa[i]]=++p;
	}
}

inline void get_h(){
	int j,k=0;
	for(int i=1;i<=n;i++){
		if(k) --k;
		j=sa[rk[i]-1];
		while(s[j+k]==s[i+k]) ++k;
		h[rk[i]]=k;
	}
}

inline void prework(){
	for(int i=1;i<=n;i++) st[i][0]=h[i];
	for(int j=1;(1<<j)<n;j++)
		for(int i=1;i+(1<<j)-1<=n;i++)
			st[i][j]=min(st[i][j-1],st[i+(1<<(j-1))][j-1]);
}

inline int lcp(int l,int r){
	if(l>r) swap(l,r); ++l;
	int x=log2(r-l+1); return min(st[l][x],st[r-(1<<x)+1][x]);
}

int root[maxn],ls[N],rs[N],sum[N],tot;

void build(int &k,int L,int R){
	k=++tot;
	if(L<R){
		int mid=(L+R)>>1;
		build(ls[k],L,mid); build(rs[k],mid+1,R);
		return;
	}
}

inline void update(int &k,int pre,int L,int R,int c){
	k=++tot;
	ls[k]=ls[pre],rs[k]=rs[pre],sum[k]=sum[pre]+1;
	if(L<R){
		int mid=(L+R)>>1;
		if(c<=mid) update(ls[k],ls[pre],L,mid,c);
		else update(rs[k],rs[pre],mid+1,R,c);
	}
}

inline int query(int u,int v,int L,int R,int ql,int qr){
	if(sum[u]-sum[v]==0) return 0;
	if(ql<=L && R<=qr) return 1;
	int mid=(L+R)>>1,ans=0;
	if(ql<=mid) ans|=query(ls[u],ls[v],L,mid,ql,qr);
	if(mid<qr) ans|=query(rs[u],rs[v],mid+1,R,ql,qr);
	return ans;
}

inline int findl(int pos,int len){
	int l=1,r=pos-1,mid,res;
	while(l<=r){
		mid=(l+r)>>1;
		if(lcp(mid,pos)>=len) res=mid,r=mid-1;
		else l=mid+1;
	} return res;
}

inline int findr(int pos,int len){
	int l=pos+1,r=n,mid,res;
	while(l<=r){
		mid=(l+r)>>1;
		if(lcp(pos,mid)>=len) res=mid,l=mid+1;
		else r=mid-1;
	} return res;
}

inline bool check(int x,int pos,int st,int ed){
	int l,r;
	if(h[rk[pos]]<x) l=rk[pos];
	else l=findl(rk[pos],x);
	if(h[rk[pos]+1]<x) r=rk[pos];
	else r=findr(rk[pos],x);
	return query(root[ed],root[st-1],1,n,l,r);
}

int main(){
	n=rd(); q=rd(); m=127; scanf("%s",s+1);
	ssort(); get_h(); prework();
	build(root[0],1,n);
	for(int i=1;i<=n;i++)
		update(root[i],root[i-1],1,n,rk[i]);
	int l,r,mid,ans;
	while(q--){
		a=rd(),b=rd(),c=rd(),d=rd();
		ans=0,l=1,r=min(b-a+1,d-c+1);
		while(l<=r){
			mid=(l+r)>>1;
			if(check(mid,c,a,b-mid+1)) ans=mid,l=mid+1;
			else r=mid-1;
		}
		printf("%d\n",ans);
	}
	return 0;
}