HDU 4622 Reincarnation（后缀数组+ST）

Description
给出一个只由小写字母组成的字符串，有m次查询，每次查询区间[l,r]这一部分子串所含有的不同子串的数量
Input
第一行为一整数T(T<=5)表示用例组数，每组用例第一行为一长度不超过2000的字符串，然后是一整数m表示查询数量(m<=10000)，之后m行每行两个整数l和r表示查询区间(1<=l<=r<=n)
Output
对于每次查询，输出查询结果
Sample Input
2
bbaba
5
3 4
2 2
2 5
2 4
1 4
baaba
5
3 3
3 4
1 4
3 5
5 5
Sample Output
3
1
7
5
8
1
3
8
5
1
Solution
类此求一个串的不同子串数量，首先对整个串做后缀数组，用ST表预处理height数组可以得到区间最小值，即得到任意两后缀的最长公共前缀，对于每次查询，O(n)按字典序枚举后缀，如果该后缀的sa值不在[l,r]中直接忽略，而在[l,r]中的后缀i，在计数时还需要满足条件且字典序在当前后缀之前的后缀pre，这两个后缀都满足条件且字典序相邻（这里的字典序指的是所有首字符在[l,r]之间的后缀在[l,r]中部分的字典序），两个后置在[l,r]之间的部分长度为la=r-sa[pre]+1和lb=r-sa[i]+1，如果la>=lb&&lcp>=lb说明pre后缀字典序在i后缀后面，不需要更新pre，否则令pre=i表示i后缀字典序在pre后缀后面（lcp=lcp(sa[i],sa[pre])），(r-l+1)*(r-l+2)/2-sum{min(lcp,min(la,lb))}即为答案
Code

#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
#define maxn 2222
int t1[maxn],t2[maxn],c[maxn],sa[maxn],Rank[maxn],height[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 str[],int n,int m)
{
    n++;
    int i,j,p,*x=t1,*y=t2;
    for(i=0;i<m;i++)c[i]=0;
    for(i=0;i<n;i++)c[x[i]=str[i]]++;
    for(i=1;i<m;i++)c[i]+=c[i-1];
    for(i=n-1;i>=0;i--)sa[--c[x[i]]]=i;
    for(j=1;j<=n;j<<=1)
    {
        p=0;
        for(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<m;i++)c[i]=0;
        for(i=0;i<n;i++)c[x[y[i]]]++;
        for(i=1;i<m;i++)c[i]+=c[i-1];
        for(i=n-1;i>=0;i--)sa[--c[x[y[i]]]]=y[i];
        swap(x,y);
        p=1;x[sa[0]]=0;
        for(i=1;i<n;i++)
        x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
        if(p>=n)break;
        m=p;
    }
    int k=0;
    n--;
    for(i=0;i<=n;i++)Rank[sa[i]]=i;
        for(i=0;i<n;i++)
        {
            if(k)k--;
            j=sa[Rank[i]-1];
            while(str[i+k]==str[j+k])k++;
            height[Rank[i]]=k;
        }
}
int RMQ[maxn],mm[maxn],best[22][maxn];
void Init_rmq(int n)
{
    for(int i=1;i<=n;i++)RMQ[i]=height[i];
    mm[0]=-1;
    for(int i=1;i<=n;i++)
        mm[i]=((i&(i-1))==0)?mm[i-1]+1:mm[i-1];
    for(int i=1;i<=n;i++)best[0][i]=i;
    for(int i=1;i<=mm[n];i++)
        for(int j=1;j+(1<<i)-1<=n;j++)
        {
            int a=best[i-1][j];
            int b=best[i-1][j+(1<<(i-1))];
            if(RMQ[a]<RMQ[b])best[i][j]=a;
            else best[i][j]=b;
        }
}
int Ask_rmq(int a,int b)
{
    int t;
    t=mm[b-a+1];
    b-=(1<<t)-1;
    a=best[t][a];b=best[t][b];
    return RMQ[a]<RMQ[b]?a:b;
}
int lcp(int a,int b)
{
    a=Rank[a];b=Rank[b];
    if(a>b)swap(a,b);
    return height[Ask_rmq(a+1,b)];
}
int T,n,m,l,r,a[maxn];
char s[maxn];
void Solve(int l,int r)
{
    int cnt=r-l+1,ans=cnt*(cnt+1)/2,pre=-1;
    for(int i=1;i<=n;i++)
    {
        if(!cnt)break;
        if(sa[i]<l||sa[i]>r)continue;
        cnt--;
        if(pre==-1)
        {
            pre=i;
            continue;
        }
        int temp=lcp(sa[pre],sa[i]);
        int la=r-sa[pre]+1,lb=r-sa[i]+1;
        if(la>=lb&&temp>=lb);
        else pre=i;
        ans-=min(temp,min(la,lb));
    }
    printf("%d\n",ans);
}
int main()
{
    scanf("%d",&T);
    while(T--)
    {
        scanf("%s",s);
        n=strlen(s);
        for(int i=0;i<n;i++)a[i]=s[i];
        a[n]=0;
        da(a,n,333);
        Init_rmq(n);
        scanf("%d",&m);
        while(m--)
        {
            scanf("%d%d",&l,&r);
            Solve(l-1,r-1);
        }
    }
    return 0;
}