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;
}