【解题报告】HDU4622-Reincarnation SA解法

后缀数组解法：这个其实和求字符串的不同子串类似，都是用所有子串的总数减去height，即重叠的部分，但是问题的关键在于全局的后缀排序不一定对应上局部的后缀排序，因此对于两个排名为pre和now的后缀，如果其局部的排名跟全局一样，则需要令pre=now；反之不一样的话就不用更新pre。这块可以这么理解：如果跟全局一样，我们已经处理了之前的pre和now，那么接下来顺着now继续就可以，反之就需要将二者交换，在这里可以直接不动pre就代表着交换，因为now被跳过去了。

然后问题就转变为如何判断局部排名是否的全局排名相同：想不出来 参考

然后还需要用ST表来处理下height数组求lcp。

AC代码：

#include <bits/stdc++.h>
using namespace std;
const int maxn=2e3+10;
typedef long long LL;
char s[maxn];
int y[maxn],x[maxn],cnt[maxn],sa[maxn],rk[maxn],height[maxn];
int n,m;
void get_SA()
{
	memset(cnt,0,sizeof(cnt));
	memset(x,0,sizeof(x));
	memset(y,0,sizeof(y));
	memset(sa,0,sizeof(sa));
	memset(rk,0,sizeof(rk));
	memset(height,0,sizeof(height));
	m=300;
	for (int i=1;i<=n;++i) ++cnt[x[i]=s[i]];
	//x[i]是第i个元素的第一关键字 
    for (int i=2;i<=m;++i) cnt[i]+=cnt[i-1]; 
    for (int i=n;i>=1;--i) sa[cnt[x[i]]--]=i; 
    for (int k=1;k<=n;k<<=1)
    {
        int num=0;
        for (int i=n-k+1;i<=n;++i) y[++num]=i;
        //y[i]表示第二关键字排名为i的数，第一关键字的位置 
        for (int i=1;i<=n;++i) if (sa[i]>k) y[++num]=sa[i]-k;
        for (int i=1;i<=m;++i) cnt[i]=0;
        for (int i=1;i<=n;++i) ++cnt[x[i]];
        for (int i=2;i<=m;++i) cnt[i]+=cnt[i-1];//第一关键字排名为1~i的数有多少个 
        for (int i=n;i>=1;--i) sa[cnt[x[y[i]]]--]=y[i],y[i]=0;
        swap(x,y);
        x[sa[1]]=1;num=1;
        for (int i=2;i<=n;++i)
            x[sa[i]]=(y[sa[i]]==y[sa[i-1]] && y[sa[i]+k]==y[sa[i-1]+k]) ? num : ++num;
        if (num==n) break;
        m=num;
    }
}
void get_height()
{
    int k=0;
    for (int i=1;i<=n;++i) rk[sa[i]]=i;
    for (int i=1;i<=n;++i)  
    {
        if (rk[i]==1) continue;//第一名height为0 
        if (k) --k;//h[i]>=h[i-1]+1;
        int j=sa[rk[i]-1];
        while (j+k<=n && i+k<=n && s[i+k]==s[j+k]) ++k;
        height[rk[i]]=k;//h[i]=height[rk[i]];
    }
}

int f[maxn][20];
void rmq_init()
{
	for(int i=1;i<=n;i++) f[i][0]=height[i];
	for(int j=1;(1<<j)<=n;j++){
		for(int i=1;i+(1<<j)-1<=n;i++){
			f[i][j]=min(f[i][j-1],f[i+(1<<(j-1))][j-1]);
		}
	}
}
int rmq_ask(int L,int R)
{
	int p=(int)(log((double(R-L+1)))/log(2.0));
	return min(f[L][p],f[R-(1<<p)+1][p]);
}
void solve(int L,int R)
{
	int pre=-1;
	int ans=(R-L+1)*(R-L+2)/2;
	for(int i=1;i<=n;i++){
		if(sa[i]<L||sa[i]>R) continue;
		if(pre==-1){
			pre=i;
			continue;
		}
		int x1=pre,x2=i;
		int lcp=rmq_ask(x1+1,x2);
		int len1=R-sa[x1]+1;
		int len2=R-sa[x2]+1;
		ans-=min(min(len1,len2),lcp);
		if(len1<=len2||lcp<len2) pre=i;
	}
	printf("%d\n",ans);
}
int main()
{
	int t;scanf("%d",&t);
	while(t--){
		scanf("%s",s+1);
		n=strlen(s+1);
		get_SA();
		get_height();
		rmq_init();
		int q;scanf("%d",&q);
		while(q--){
			int L,R;scanf("%d%d",&L,&R);
			solve(L,R);
		}
	}
}