[后缀数组+枚举] hdu 4622 Reincarnation

题意：求一个字符串中，[a,b]内的不同子串数是多少。

思路：跟之前的求整个字符串中的不同子串数的方法一样，总的个数减去重复个数。

当然这题显然不能截取一段做一次DA，这样是超时的。

当然也不能选完在排序，我们其实可以种一个方法规避排序。

首先我们知道sa是排好序的了，那么我们只要标记一下[a,b]之间的ra[i]，也就是sa里的位置。

这样我们遍历一遍sa，计算一下那些被标记的位置的lcp就好了！

代码：

#include"cstdlib"
#include"cstdio"
#include"cstring"
#include"cmath"
#include"queue"
#include"algorithm"
#include"iostream"
#include"map"
using namespace std;
#define N 20020
int wa[N],wb[N],wv[N],wws[N];
int sa[N],ra[N],height[N],Log[N];
int v[N];
int 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)
{
    int i,j,p,*x=wa,*y=wb;
    for(i=0; i<m; i++) wws[i]=0;
    for(i=0; i<n; i++) wws[x[i]=v[i]]++;
    for(i=1; i<m; i++) wws[i]+=wws[i-1];
    for(i=n-1; i>=0; i--) sa[--wws[x[i]]]=i;
    for(j=1,p=1; p<n; j*=2,m=p)
    {
        for(i=n-j,p=0; 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++) wws[i]=0;
        for(i=0; i<n; i++) wws[wv[i]]++;
        for(i=1; i<m; i++) wws[i]+=wws[i-1];
        for(i=n-1; i>=0; i--) sa[--wws[wv[i]]]=y[i];
        for(swap(x,y),p=1,i=1,x[sa[0]]=0; i<n; i++) x[sa[i]]=cmp(y,sa[i],sa[i-1],j)?p-1:p++;
    }
    return ;
}
void gethei(int n)
{
    int i,j,k=0;
    for(i=1; i<=n; i++) ra[sa[i]]=i;
    for(i=0; i<n; i++)
    {
        if(k) k--;
        j=sa[ra[i]-1];
        while(v[i+k]==v[j+k])
            k++;
        height[ra[i]]=k;
    }
    return ;
}
char fuck[12345];
int dp[N][30];
void rmqinit(int n)
{
    int m=Log[n];
    int i,j;
    for(i=1; i<=n; i++) dp[i][0]=height[i];
    for(i=1; i<=m; i++)
    {
        for(j=1; j+(1<<i)-1<=n; j++)
            dp[j][i]=min(dp[j][i-1],dp[j+(1<<i>>1)][i-1]);
    }
    return ;
}
int lcp(int a,int b)
{
    a=ra[a];
    b=ra[b];
    if(a>b) swap(a,b);
    a++;
    int m=Log[b-a+1];
    return min(dp[a][m],dp[b-(1<<m)+1][m]);
}
int main()
{
    int t,i;
    Log[0]=-1;
    for(i=1; i<=N; i++)
        Log[i]=(i&(i-1))?Log[i-1]:Log[i-1]+1;
    cin>>t;
    while(t--)
    {
        scanf("%s",fuck);
        int n=strlen(fuck);
        for(i=0; i<n; i++) v[i]=fuck[i];
        v[n]=0;
        da(n+1,550);
        gethei(n);
        rmqinit(n);
        int q;
        cin>>q;
        while(q--)
        {
            int x,y;
            scanf("%d%d",&x,&y);
            int ans=(y-x+1)*(1+y-x+1)/2;  //总的长度
            int tep[2234];
            for(i=0; i<=n; i++) tep[i]=0;  //标记清零
            for(i=x-1; i<=y-1; i++) tep[ra[i]]=1;  //标记哪些出现过
            int last=-1;  //因为第一个不算 标记是否是第一个
            int d=0;      //d表示在当前位置之前的最长的公共子串长度
            for(i=1; i<=n; i++)  //遍历sa
            {
                if(tep[i])
                {
                    if(last!=-1)
                    {
                        int z=lcp(sa[last],sa[i]);
                        d=min(d,z);       //对于所能提供的长度，d和lcp取最小
                        d=max(d,min(z,y-sa[last]));  //对于最长的长度，取d与（lcp与前一个后缀长度的最小值）的最大值
                        ans-=min(d,y-sa[i]);        //重复的则是 d与当前长度的最小值
                    }
                    last=i;
                }
            }
            printf("%d\n",ans);
        }
    }
    return 0;
}