Light OJ 1314 Names for Babies

http://www.lightoj.com/volume_showproblem.php?problem=1314

题意:给定一个串和p,q，求长度在p到q之间的子串有几种

思路:后缀数组，对于每个位置的贡献是min(n-sa[i],q)，然后要减去重复和没有的部分，就是max(height[i],p-1)，当然，还要对0取个max

#include<cstdio>
#include<iostream>
#include<cmath>
#include<cstring>
#include<algorithm>
char s[200005];
int num[200005],sa[200005],rank[200005],h[200005],p,q,n;
int ws[200005],wa[200005],wb[200005],wv[200005];
int read(){
    int t=0,f=1;char ch=getchar();
    while (ch<'0'||ch>'9'){if (ch=='-')f=-1;ch=getchar();}
    while ('0'<=ch&&ch<='9'){t=t*10+ch-'0';ch=getchar();}
    return t*f;
}
void solve(){
    int ans=0;
    for (int i=1;i<=n;i++)
     ans+=std::max(0,std::min(q,n-sa[i])-std::max(p-1,h[i]));
    printf("%d
",ans); 
}
bool cmp(int *r,int a,int b,int l){
    return r[a]==r[b]&&r[a+l]==r[b+l];
}
void da(int *r,int *sa,int n,int m){
    int *x=wa,*y=wb,*t,i,j,p;
    for (i=0;i<n;i++) x[i]=r[i];
    for (i=0;i<m;i++) ws[i]=0;
    for (i=0;i<n;i++) ws[x[i]]++;
    for (i=1;i<m;i++) ws[i]+=ws[i-1];
    for (i=n-1;i>=0;i--) sa[--ws[x[i]]]=i;
    for (j=1,p=1;p<n;m=p,j*=2){
        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<n;i++) wv[i]=x[y[i]];
        for (i=0;i<m;i++) ws[i]=0;
        for (i=0;i<n;i++) ws[wv[i]]++;
        for (i=1;i<m;i++) ws[i]+=ws[i-1];
        for (i=n-1;i>=0;i--) sa[--ws[wv[i]]]=y[i];
        for (t=x,x=y,y=t,i=1,x[sa[0]]=0,p=1;i<n;i++)
         x[sa[i]]=cmp(y,sa[i-1],sa[i],j)?p-1:p++;
    }
}
void cal(int *r,int n){
    int i,j,k=0;
    for (i=1;i<=n;i++) rank[sa[i]]=i;
    for (i=0;i<n;h[rank[i++]]=k)
     for (k?k--:0,j=sa[rank[i]-1];r[i+k]==r[j+k];k++);
}
int main(){
    int T=read(),Tcase=0;
    while (T--){
        Tcase++;printf("Case %d: ",Tcase);
        scanf("%s",s);
        p=read();q=read();
        n=strlen(s);
        for (int i=0;i<n;i++) num[i]=s[i];
        num[n]=0;
        da(num,sa,n+1,200);
        cal(num,n);
        solve();
    }
}