题目链接:http://www.lightoj.com/volume_showproblem.php?problem=1314
题意:求串A长度在[p,q]区间上的子串个数。
思路:求sa和height数组。每个位置sa[i]最多产生n-sa[i]个子串,min(q,n-sa[i])为实际产生的子串,其中有height[i]个重的,减去。
int r[N],sa[N],wa[N],wb[N],wd[N],rank[N],h[N];
int cmp(int *r,int a,int b,int len)
{
return r[a]==r[b]&&r[a+len]==r[b+len];
}
void da(int *r,int *sa,int n,int m)
{
int i,j,p,*x=wa,*y=wb,*t;
FOR0(i,m) wd[i]=0;
FOR0(i,n) wd[x[i]=r[i]]++;
FOR1(i,m-1) wd[i]+=wd[i-1];
FORL0(i,n-1) sa[--wd[x[i]]]=i;
for(j=1,p=1;p<n;j<<=1,m=p)
{
p=0;
FOR(i,n-j,n-1) y[p++]=i;
FOR0(i,n) if(sa[i]>=j) y[p++]=sa[i]-j;
FOR0(i,m) wd[i]=0;
FOR0(i,n) wd[x[i]]++;
FOR1(i,m-1) wd[i]+=wd[i-1];
FORL0(i,n-1) sa[--wd[x[y[i]]]]=y[i];
t=x;x=y;y=t;p=1;x[sa[0]]=0;
FOR1(i,n-1) x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
}
}
void calHeight(int *r,int *sa,int n)
{
int i,j,k=0;
FOR1(i,n) rank[sa[i]]=i;
FOR0(i,n)
{
if(k) k--;
j=sa[rank[i]-1];
while(i+k<n&&j+k<n&&r[i+k]==r[j+k]) k++;
h[rank[i]]=k;
}
}
char s[N];
int n,p,q;
i64 deal()
{
i64 ans=0,i;
FOR1(i,n) ans+=max(0,min(q,n-sa[i])-max(p-1,h[i]));
return ans;
}
int main()
{
int C,num=0;
RD(C);
while(C--)
{
RD(s);
RD(p,q);
int i;
n=strlen(s);
FOR0(i,n) r[i]=s[i];
r[n]=0;
da(r,sa,n+1,130);
calHeight(r,sa,n);
printf("Case %d: %lld\n",++num,deal());
}
return 0;
}