相似子串 BZOJ 3230 后缀数组模板题

相似子串 BZOJ 3230 后缀数组模板题

时间2019 8 10——这题网上似乎找不到了，但是我有数据哟，大家可以下载
附上链接https://download.csdn.net/my/uploads

这道题其实非常简单，但是我场上并没有想出来。。。。

现在拿出来补补智商

首先我们求出后缀数组sa[] height[] Rank[] 数组，到这我们的基本条件就已经具备了

然后再根据我们对后缀数组的理解：两个排名不相邻的最长公共前缀定义为排名在它们之间的Height的最小值

RMQ问题当然首选ST表啊

那么我们怎么来获得不同本质的子串呢？

把子串全部求出来是不可能的，这时候我们就又用到了后缀数组的height [] 数组

两个排名相邻的数组，它们的LCP以及其前缀是肯定不能重复的，所以我们就只需要记录每一个后缀对不同本质子串的贡献即可，即该后缀的长度减去LCP的长度，我们用一个num数组记录贡献的前缀和，后面我们能使用二分(lowwer_bound)查找位置。

最后正反求最长公共前后缀的长度即可。

但是我傻逼，忘记了自己的sa[]数组记录的是从0开始的位置、rank数组是纪录第i位置后缀的排名。。。然后改了两天都没发现这个错误。。。。。

是我太菜了，不配写题

如果我的题解看不懂：在这里推荐另一个博主的博客，大家可以去膜拜一下https://www.cnblogs.com/yoyoball/p/9324662.html

#include <bits/stdc++.h>
#define ll long long
using namespace std;
const int maxn=1e5+7;
int len,m;
char str[maxn];
struct suf{
	int ST[maxn][20];
	int wa[maxn],wb[maxn],wv[maxn],wt[maxn];
	int r[maxn];
	int sa[maxn],height[maxn],Rank[maxn];
	ll num[maxn];//记录每一个子后缀对子串的贡献
	bool cmp(int *r,int a,int b,int l){
		return r[a]==r[b]&&r[a+l]==r[b+l];
	}
	void da(int n,int m){
		for(int i=0;i<len;i++) r[i]=str[i];
		r[len]=0;
		int i,j,p,*x=wa,*y=wb,*t;
		for(i=0;i<m;i++) wt[i]=0;
		for(i=0;i<n;i++) wt[x[i]=r[i]]++;
		for(i=1;i<m;i++) wt[i]+=wt[i-1];
		for(i=n-1;i>=0;i--) sa[--wt[x[i]]]=i;
		for(j=1,p=1;p<n;j*=2,m=p){
			for(p=0,i=n-j;i<n;i++) y[p++]=i;
			for(i=0;i<n;i++) if(sa[i]>=j) y[p++]=sa[i]-j;
			for(i=0;i<n;i++) wv[i]=x[y[i]];
			for(i=0;i<m;i++) wt[i]=0;
			for(i=0;i<n;i++) wt[wv[i]]++;
			for(i=1;i<m;i++) wt[i]+=wt[i-1];
			for(i=n-1;i>=0;i--) sa[--wt[wv[i]]]=y[i];
			for(t=x,x=y,y=t,p=1,x[sa[0]]=0,i=1;i<n;i++)
				x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++; 
		}
		calheight();
	}
	void calheight(){
		int i,j,k=0; 
		for( i=1;i<=len;i++ ) Rank[sa[i]]=i;  
		for(i=0;i<len;i++) {  
			if(k)k--;  
			int j=sa[Rank[i]-1];  
			while(r[i+k]==r[j+k]) k++; 
			height[Rank[i]]=k; 
		}     
		getnum();
		RMQ();
	}
	void getnum(){
		num[0]=0;
		for(int i=1;i<=len;i++)
			num[i]=num[i-1]+(len-sa[i]-height[i]);//先计算排名靠前的
							//先求得本身长度再减去与上一个排名的LCP即为对不同子串的贡献
	}
	void RMQ(){
		for(int i=1;i<=len;i++)
			ST[i][0]=height[i];
		for(int i=1;(1<<i)<=len;i++)
			for(int j=len-(1<<i)+1;j>=1;j--)
				ST[j][i]=min(ST[j][i-1],ST[j+(1<<i-1)][i-1]);
	}
	int lcp(int x,int y){
		if (x==y) return len-sa[x];
        if (x>y) swap(x,y);
        ++x;
        int tmp=(int)(log(1.0*(y-x+1))/log(2.0));
        return min(ST[x][tmp],ST[y-(1<<tmp)+1][tmp]);//获取两个串的位置之间最小的最长公共前缀，即这两个串最多能匹配到的最长公共前缀
	}
	int query(ll x,ll y,int &edx,int &edy,int &lenx,int &leny){
		int posx,posy;
        posx=lower_bound(num+1,num+1+len,x)-num;
        posy=lower_bound(num+1,num+1+len,y)-num;//获取字典序为x和为y的子串的后缀串所处的位置
		int ret=lcp(posx,posy);
		lenx=len-sa[posx]-(num[posx]-x);
        leny=len-sa[posy]-(num[posy]-y);//分别获取两个字串的长度
        edx=len-sa[posx]-lenx;
        edy=len-sa[posy]-leny;//两个子串结尾的位置
        return min(ret,min(lenx,leny));//返回能匹配到的最长公共前缀的长度
	}
	int query2(int stx,int sty,int lenx,int leny){
        int ret=lcp(Rank[stx],Rank[sty]);
        return min(ret,min(lenx,leny));
    }
}a[2];
void pre(){
    a[0].da(len+1,300);
	strrev(str);
    a[1].da(len+1,300);
}
int main(){
//	freopen(".in","r",stdin);
//	freopen(".out","w",stdout);
    scanf("%d %d",&len,&m);
    scanf("%s",str);
	pre();
	// for(int i=1;i<=len;i++)
	// 	printf("%d: %d\n",i,a[0].sa[i]);
	while(m--){
		int tmp1,tmp2,edx,edy,lenx,leny;
		ll x,y;
		scanf("%lld %lld",&x,&y);
		if (x>a[0].num[len] || y>a[0].num[len])
			printf("-1\n");
		else {
			tmp1=a[0].query(x,y,edx,edy,lenx,leny);//正向
            tmp2=a[1].query2(edx,edy,lenx,leny);//反向
            printf("%lld\n",1LL*tmp1*tmp1+1LL*tmp2*tmp2);
		}
	}
return 0;
}