HDU 4622 后缀数组+RMQ

点击打开链接

题意：给一个字符串，有q次询问，为你从第L个到第R个字符组成的不同的子串的数量

思路：做过好多这种提问的了，减去的数量就是lcp[i]的值，不过这个是一个区间内的操作，正解是后缀自动机，然而还没学~~~，发现对于n*q*log应该没什么问题，就写了写，对于询问的这个区间，我一开始写的是跑sa数组，如果sa[i]在L和R区间内，那么加进去，与下一个在这个区间的sa进行RMQ查询最长公共前缀，更新这个两个值即可，还要注意的是虽然最长公共前缀可能很大但是这个区间没那么大，就需要处理一下了，过了样例交了一发WA，在交WA，在交WA.......好吧思路应该是错了，看了网上题解，发现写的差不多嘛（(/ □ \)），其实思路完全不一样，那么错在哪呢，是因为对于现在查询的区间它的sa数组和整个大区间的sa数组顺序可能是不一样的，那么如何用整体的来调整区间的就是任务了，举个例子

如：abacab 整体的sa为 5 ab     1 abacab    3 acab       6 b     2 bacab      4   cab

而我们如果要[1,3]的，那么这部分的sa为   3 a      1 aba      2 ba  那么有一部分错了即3与1的位置，那么该如何调整呢，就是代码中注释那句话，当abacab与acab比较时，在部分中的lcp为1，而对于部分来说，lcp大于3到最后的长度，则说明它的sa肯定是比aba高的，这很好懂吧，所以不需要更新pos，依次类推下去就可以了

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <iostream>
#include <algorithm>
using namespace std;
typedef long long ll;
typedef unsigned long long ull;
const int inf=0x3f3f3f3f;
const ll INF=0x3f3f3f3f3f3f3f3fll;
const int MAXN=4010;
int wa[MAXN],wb[MAXN],wv[MAXN],ww[MAXN];
int sa[MAXN],lcp[MAXN],Rank[MAXN],rank1[MAXN],dp[MAXN][20];
char str[MAXN];
inline bool cmp(int *r,int a,int b,int len){
    return r[a]==r[b]&&r[a+len]==r[b+len];
}
 void construct_sa(int n,int m){
     int i,j,p,*x=wa,*y=wb,*t;n++;
     for(i=0;i<m;i++) ww[i]=0;
     for(i=0;i<n;i++) ww[x[i]=str[i]]++;
     for(i=1;i<m;i++) ww[i]+=ww[i-1];
     for(i=n-1;i>=0;i--) sa[--ww[x[i]]]=i;
     for(j=p=1;p<n;j<<=1,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<m;i++) ww[i]=0;
         for(i=0;i<n;i++) ww[wv[i]=x[y[i]]]++;
         for(i=1;i<m;i++) ww[i]+=ww[i-1];
         for(i=n-1;i>=0;i--) sa[--ww[wv[i]]]=y[i];
         for(t=x,x=y,y=t,x[sa[0]]=0,p=i=1;i<n;i++)
             x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
     }
}
void construct_lcp(int n){
    for(int i=0;i<=n;i++) rank1[sa[i]]=i;
    int h=0;
    lcp[0]=0;
    for(int i=0;i<n;i++){
        int j=sa[rank1[i]-1];
        if(h>0) h--;
        for(;j+h<n&&i+h<n;h++) if(str[i+h]!=str[j+h]) break;
        lcp[rank1[i]-1]=h;
    }
}
void RMQ_init(int n){
    for(int i=1;i<=n;i++)
        dp[i][0]=lcp[i-1];
    for(int i=1;(1<<i)<=n;i++){
        for(int j=1;j+(1<<i)-1<=n;j++){
            dp[j][i]=min(dp[j][i-1],dp[j+(1<<(i-1))][i-1]);
        }
    }
}
int RMQ(int le,int ri){
    le=rank1[le];ri=rank1[ri];
    if(le>ri) swap(le,ri);le++;
    int k=0;
    while((1<<(k+1))<=ri-le+1) k++;
    int ans2=min(dp[le][k],dp[ri-(1<<k)+1][k]);
    return ans2;
}
int main(){
    int T,m,a,b;
    scanf("%d",&T);
    while(T--){
        scanf("%s",str);
        int len=strlen(str);
        construct_sa(len,120);
        construct_lcp(len);
        RMQ_init(len);
        scanf("%d",&m);
        while(m--){
            scanf("%d%d",&a,&b);
            a--;b--;
            int tmp=b-a+1,pos=-1,now;
            int ans=tmp*(tmp+1)/2;
            for(int i=0;i<=len;i++){
                if(sa[i]<=b&&sa[i]>=a){
                    if(pos==-1) pos=sa[i];
                    else{
                        now=sa[i];
                        int kk=RMQ(now,pos);
                        int tt=min(kk,min(b-now,b-pos)+1);
                        ans-=tt;
                        if(pos<now&&kk>=(b-now+1)){}//和我自己写的就差这么一步，o(︶︿︶)o 唉
                        else pos=sa[i];
                    }
                }
            }
            printf("%d\n",ans);
        }
    }
    return 0;
}