[poj3261]Milk Patterns

求出后缀数组和height数组，然后二分答案后分组查询，一个块内如果有超过k个那么这个答案就可以。

#include<bits/stdc++.h>
using namespace std;
#define N 50005
int n,m,ans,a[N],b[N],h[N],sum[N],ra[N<<1],sa[N];
char s[N],s1[N];
void change(){
    for(int i=0;i<n;i++)a[i]=s[i];
    memcpy(ra,a,sizeof(a)); 
    sort(a,a+n);
    m=unique(a,a+n)-a;
    for(int i=0;i<n;i++)ra[i]=lower_bound(a,a+m,ra[i])-a+1;
}
void sort(int k){
    memset(sum,0,sizeof(sum));
    for(int i=0;i<n;i++)sum[ra[i+k]+1]++;
    for(int i=1;i<=m;i++)sum[i+1]+=sum[i];
    for(int i=0;i<n;i++)b[sum[ra[a[i]+k]]++]=a[i];
}
void build(){
    for(int i=1;;i<<=1){
        for(int j=0;j<n;j++)a[j]=j;
        sort(i);
        memcpy(a,b,sizeof(a));
        sort(0);
        m=a[b[0]]=1;
        for(int j=1;j<n;a[b[j++]]=m)
            if ((ra[b[j-1]]!=ra[b[j]])||(ra[b[j-1]+i]!=ra[b[j]+i]))m++; 
        memcpy(ra,a,sizeof(a));
        if (m==n)break;
    }
    for(int i=0;i<n;i++)sa[ra[i]]=i;
}
void height(){
    m=0;
    for(int i=0;i<n;h[ra[i++]]=m,m-=(m>0))
        for(int j=sa[ra[i]-1];(i!=j)&&(s[i+m]==s[j+m]);m++);
    h[1]=0;
}
int lcp(int x,int y){
    int ans=0x3f3f3f3f;
    if (ra[x]>ra[y])swap(x,y);
    for(int i=ra[x]+1;i<=ra[y];i++)ans=min(ans,h[i]);
    return ans;
}
void write(int x,int y){
    for(int i=x;i<=y;i++)printf("%c",s[i]);
}
int main(){
    int t;
    scanf("%d",&t);
    while (t--){
        scanf("%s",s);
        n=strlen(s);
        change();
        build();
        height();
        ans=0;
        for(int i=1;i<=n;i++)ans+=n-sa[i]-h[i];
        printf("%d\n",ans);
    }
}